Update app.py

app.py

@@ -60,10 +60,8 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=
         "output_type":"pil",
     }
     
-    images = model(**parameters).images
-    print(images)
-    image_paths = [save_image(img) for img in images]
-    return image_paths
+    image = model(**parameters).images[0]
+    return image
 
 # Initialize
 with gr.Blocks() as main: