Update app.py

app.py

@@ -66,10 +66,17 @@ torch.hub.download_url_to_file('https://hai.stanford.edu/sites/default/files/sty
 model = torch.hub.load('pytorch/vision:v0.6.0', 'deeplabv3_resnet101', pretrained=True)
 model.eval()
 
+inputs_image = [
+    gr.components.Image(type="filepath", label="Input Image"),
+]
+outputs_image = [
+    gr.components.Image(type="numpy", label="Output Image"),
+]
+
 gr.Interface(
-    inference,
-    gr.inputs.Image(type="pil", label="Input"),
-    gr.outputs.Image(type="pil", label="Output"),
+    fn=inference,
+    inputs=inputs_image,
+    outputs=outputs_image,
     title=title,
     description=description,
     article=article,