Update app.py

app.py

@@ -2,7 +2,8 @@ import os
 import torch
 import spaces
 import gradio as gr
-from diffusers import FluxFillPipeline, FluxTransformer2DModel, AutoencoderKL
+from diffusers import FluxFillPipeline
+from diffusers.utils import load_image
 import random
 import numpy as np
 from huggingface_hub import hf_hub_download
@@ -33,15 +34,18 @@ def inpaintGen(
         randomize_seed: bool,
         progress=gr.Progress(track_tqdm=True)):
 
-    source_img = imgMask["background"]
-    mask_img = imgMask["layers"][0]
+    source_path = imgMask["background"]
+    mask_path = imgMask["layers"][0]
 
-    if not source_img:
+    if not source_path:
         raise gr.Error("Please upload an image.")
 
-    if not mask_img:
+    if not mask_path:
         raise gr.Error("Please draw a mask on the image.")
-        
+
+    source_img = load_image(source_path)
+    mask_img = load_image(mask_path)
+    
     width, height = source_img.size
 
     if randomize_seed:
@@ -74,7 +78,7 @@ with gr.Blocks(theme="ocean", title="Flux.1 dev inpaint", css=CSS) as demo:
     """)
     with gr.Row():
         with gr.Column():
-            imgMask = gr.ImageMask(type="pil", label="Image", layers=False, height=800)
+            imgMask = gr.ImageMask(type="filepath", label="Image", layers=False, height=800)
             inpaint_prompt = gr.Textbox(label='Prompts ✏️', placeholder="A hat...")
             with gr.Row():
                 Inpaint_sendBtn = gr.Button(value="Submit", variant='primary')