Update worker_runpod.py

worker_runpod.py

@@ -191,11 +191,12 @@ def generate(input):
             samples.clamp_(0.0, 1.0)
 
             img = to_pil_image(samples[0].float())
+            img.save("/content/out.png")
 
         except Exception:
             print(traceback.format_exc())
 
-    result = img
+    result = "/content/out.png"
     response = None
     try:
         source_id = values['source_id']