@@ -84,6 +84,11 @@ if __name__ == '__main__':
        if i % 5 == 0:  # save images to an HTML file
            print('processing (%04d)-th image... %s' % (i, img_path))
        save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio, width=opt.display_winsize)
+
+        model.fake = None
+        model.real = None
+        model.fake_noattack = None
+        
    webpage.save()  # save the HTML