update app.py

app.py

@@ -30,7 +30,7 @@ if not os.path.exists("./models/masksupnyu39.31d.pth"):
 
 # Load model
 model = NestedUNet(num_classes=40)
-checkpoint = torch.load("./models/masksupnyu39.31d.pth")
+checkpoint = torch.load("./models/masksupnyu39.31d.pth", map_location=torch.device('cpu'))
 model.load_state_dict(checkpoint)
 model = model.to(DEVICE)
 model.eval()
@@ -64,8 +64,8 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2210.009
 
 gr.Interface(
     inference,
-    gr.inputs.Image(type='file', label="Input Image"),
-    gr.outputs.Image(type="file", label="Predicted Output"),
+    gr.inputs.Image(type='filepath', label="Input Image"),
+    gr.outputs.Image(type="filepath", label="Predicted Output"),
     examples=["./sample_images/a.png", "./sample_images/b.png", 
     "./sample_images/c.png", "./sample_images/d.png"],
     title=title,