Update app.py

app.py

@@ -98,7 +98,6 @@ def generate(
             output_type="pil",
         ).images
     elif model == "Alpha inpaint":
-        print("Inpainting...")
         blurred_mask = pipe_inpaint.mask_processor.blur(mask_image, blur_factor=blur_factor)
         images = pipe_inpaint(
             prompt=prompt,
@@ -114,7 +113,9 @@ def generate(
             output_type="pil",
         ).images
 
-    return [images[0]], seed
+    image_paths = [save_image(img) for img in images]
+    print(image_paths)
+    return image_paths, seed
 
         
 
@@ -246,6 +247,11 @@ with gr.Blocks(title="Visionix Playground", css=css) as demo:
         ],
         fn=generate,
         inputs=[
+            model,
+            inpaint_image,
+            mask_image,
+            blur_factor,
+            strength,
             prompt,
             negative_prompt,
             use_negative_prompt,