Update main.py

main.py

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