Update app.py
Browse files
app.py
CHANGED
@@ -67,10 +67,10 @@ model = torch.hub.load('pytorch/vision:v0.6.0', 'deeplabv3_resnet101', pretraine
|
|
67 |
model.eval()
|
68 |
|
69 |
inputs_image = [
|
70 |
-
gr.components.Image(type="
|
71 |
]
|
72 |
outputs_image = [
|
73 |
-
gr.components.Image(type="
|
74 |
]
|
75 |
|
76 |
gr.Interface(
|
|
|
67 |
model.eval()
|
68 |
|
69 |
inputs_image = [
|
70 |
+
gr.components.Image(type="pil", label="Input Image"),
|
71 |
]
|
72 |
outputs_image = [
|
73 |
+
gr.components.Image(type="pil", label="Output Image"),
|
74 |
]
|
75 |
|
76 |
gr.Interface(
|