Update app.py

app.py

@@ -49,10 +49,14 @@ class ADA_SKIN(object):
         return
 
     def _predict_image(self, img, cat):
+        # Convertir la imagen a PILImage
+        img = PILImage.create(img)
         pred, idx, probs = learn.predict(img)
         return dict(zip(cat, map(float, probs)))
 
     def _predict_image2(self, img, cat):
+        # Convertir la imagen a PILImage
+        img = PILImage.create(img)
         pred, idx, probs = learn2.predict(img)
         return dict(zip(cat, map(float, probs)))
 
@@ -121,7 +125,7 @@ maxi.categories = learn.dls.vocab
 maxi.categories2 = learn2.dls.vocab
 
 # Crear la interfaz de Gradio
-hf_image = gr.Image(type='pil')  # Se elimina el argumento 'shape'
+hf_image = gr.Image()  # Se elimina el argumento 'type' porque no es necesario
 hf_plot = gr.Plot()
 
 intf = gr.Interface(fn=maxi.predict_donut,