Adjustments

app.py

@@ -151,12 +151,7 @@ def optimize_network(
     #     np.random.seed(seed)
     #     torch.manual_seed(seed)
     #     random.seed(seed)
-
-    layer = int(layer)
-    cutn = int(cutn)
-    num_iterations = int(num_iterations)
-    neuron = int(neuron)
-    display_rate = int(display_rate)
+    save_progress_video = True
     
     make_cutouts = MakeCutouts(clip_models[clip_model].visual.input_resolution, cutn)
     loss_fn = CLIPActivationLoss(clip_models[clip_model].visual.transformer.resblocks[layer],
@@ -240,7 +235,12 @@ def inference(
     maximize,
     display_rate = 20
 ):
-
+    layer = int(layer)
+    cutn = int(cutn)
+    num_iterations = int(num_iterations)
+    neuron = int(neuron)
+    display_rate = int(display_rate)
+    
     opt_type = 'MADGRAD'
     seed = 20
     save_progress_video = True