Update app.py

app.py

@@ -4,26 +4,26 @@ import skimage
 
 learn = load_learner('model.pkl')
 
-labels = learn.dls.vocab
-
 def predict(img):
     img = PILImage.create(img)
+    img = img.resize((224, 224))  # Redimensionar imagen aquí si es necesario
     pred, pred_idx, probs = learn.predict(img)
     return pred
 
 title = "Modelo de Predicción Aves PARCIAL PIB"
 description = "Este modelo utiliza redes neuronales para clasificar diferentes especies de aves en imágenes. En la parte inferior, puedes probar el modelo con algunas imágenes de muestra."
 examples = ['3.jpg', '5.jpg', 'images.jpeg', '1.jpg', '2.jpg', '4.jpg', '6.jpg']
-enable_queue = True
 
-gr.Interface(
+# Crear la interfaz
+interface = gr.Interface(
     fn=predict,
     inputs=gr.Image(),
     outputs=gr.Label(num_top_classes=1),
     title=title,
     description=description,
     examples=examples,
-    enable_queue=enable_queue,
-    layout="vertical",
-    thumbnail=None
-).launch()
+    layout="vertical"
+)
+
+# Lanzar la interfaz
+interface.launch()