Update app.py

app.py

@@ -12,14 +12,13 @@ import imageio
 os.mkdir("data")
 os.rename("best.ckpt", "models/best.ckpt")
 os.mkdir("dataout")
-model = hub.Module(name='U2Net')
 
 
-def infer(img, mask):
+def infer(img):
     
     
     img = Image.open(img)
-    mask = Image.open(mask)
+    mask = Image.open("./masks/modelscope-mask.png")
     inverted_mask = ImageOps.invert(mask)
     
     
@@ -29,8 +28,7 @@ def infer(img, mask):
     return "./dataout/data_mask.png", "./data/data_mask.png"
 
 
-inputs = [gr.Image(label="Input", source="upload", type="filepath"),
-          gr.Image(label="Mask", source="upload", type="filepath")]
+inputs = [gr.Image(label="Input", source="upload", type="filepath")]
 outputs = [gr.outputs.Image(type="file", label="output"),
            gr.outputs.Image(type="file", label="Mask")]
 title = "LaMa Image Inpainting"