Update app.py
Browse files
app.py
CHANGED
@@ -35,4 +35,4 @@ def predict(img):
|
|
35 |
mask=np.reshape(mask,(480,640))
|
36 |
return Image.fromarray(mask.astype('uint8'))
|
37 |
|
38 |
-
gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(128, 128)), outputs=
|
|
|
35 |
mask=np.reshape(mask,(480,640))
|
36 |
return Image.fromarray(mask.astype('uint8'))
|
37 |
|
38 |
+
gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(128, 128)), outputs=[gr.outputs.Image(type="pil", label="Predicci贸n")], examples=['color_155.jpg','color_154.jpg']).launch(share=False)
|