update requirements.txt

app.py

@@ -47,7 +47,7 @@ def generate(image_mask, prompt, num_inference_steps=50, image_guidance_scale=1.
         generator=generator
     ).images[0]
 
-    return image,mask_img
+    return image
 
 
 # image_mask_input = gr.ImageMask(label="Input Image", type="pil", brush_color="#000000", elem_id="inputmask",

requirements.txt

@@ -4,4 +4,5 @@ torch
 transformers
 xformers
 sentencepiece
--e UltraEdit/diffusers
+-e UltraEdit/diffusers
+gradio==4.29.0