Update app.py

app.py

@@ -34,6 +34,7 @@ def predict(img):
     mask=np.reshape(mask,(400,400))
     return Image.fromarray(mask.astype('uint8')
     
-# Creamos la interfaz y la lanzamos. 
-gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(400, 400)), outputs="image", examples=['examples/1CA SUR_1200_800.png', 'examples/1CA SUR_4000_1200.png', 'examples/1CA SUR_4800_2000.png']).launch(share=False)
+# Creamos la interfaz y la lanzamos.
+#examples=['examples/1CA SUR_1200_800.png', 'examples/1CA SUR_4000_1200.png', 'examples/1CA SUR_4800_2000.png']
+gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(400, 400)), outputs=gr.outputs.Image(type="pil")).launch(share=False)