Update app.py

app.py

@@ -1,4 +1,4 @@
-from icevision.all import ClassMap, tfms, models
+#from icevision.all import *
 import torch
 import gradio as gr
 
@@ -15,4 +15,4 @@ def predict(img):
     return img
     
 # Creamos la interfaz y la lanzamos. 
-gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(128, 128)), outputs=gr.outputs.Image(label='Imagen resultado'),examples=['00001.jpg','00002.jpg']).launch(share=False)
+gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(128, 128)), outputs=gr.outputs.Image(type="pil",label='Imagen resultado'),examples=['00001.jpg','00002.jpg']).launch(share=False)