Update main.py

main.py

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