Create app.py

app.py

@@ -0,0 +1,20 @@
+import gradio as gr
+from fastai.vision.all import *
+
+# Carga tu modelo
+learn = load_learner("clasificar_enemigos_BOTW.pkl")  # Reemplaza con la ruta a tu modelo
+
+# Define la función de predicción
+def predict(img):
+    img = PILImage.create(img)
+    pred, pred_idx, probs = learn.predict(img)
+    return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(learn.dls.vocab))}
+
+# Crea la interfaz de Gradio
+gr.Interface(
+    fn=predict,
+    inputs=gr.Image(),
+    outputs=gr.Label(num_top_classes=2),
+    title="Clasificador de Enemigos de BOTW",
+    description="Sube una imagen de un enemigo de BOTW para clasificarlo."
+).launch()