Update main.py

main.py

@@ -519,10 +519,10 @@ def main_func(target, experiment, cfg_arg):
                     imshow = img.detach().cpu()
                 pydiffvg.imwrite(imshow, filename, gamma=gamma)
 
-                ### added for app
-                if int(t)%30==0:
-                    print(f"debug: {t}, {filename} {img.size()}")
-                    return img.detach().cpu().numpy(), int(t)
+            ### added for app
+            if int(t)%30==0:
+                print(f"debug: {t}, {filename} {img.size()}")
+                return img.detach().cpu().numpy(), int(t)
 
             x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW