Update app.py

app.py

@@ -29,7 +29,7 @@ def resize(height,img):
   img = img.resize((wsize,baseheight), Image.Resampling.LANCZOS)
   return img
 
-def predict(source_img):
+def predict(prompt, source_img):
     imageio.imwrite("data.png", source_img["image"])
     imageio.imwrite("data_mask.png", source_img["mask"]) 
     
@@ -50,4 +50,4 @@ def predict(source_img):
 
 custom_css="style.css"
 
-gr.Interface(fn=predict, inputs=source_img, outputs=gallery, css=custom_css).launch(enable_queue=True)
+gr.Interface(fn=predict, inputs=["text", source_img], outputs=gallery, css=custom_css).launch(enable_queue=True)