Create app.py

app.py

@@ -0,0 +1,23 @@
+import gradio as gr
+from fastai.vision.all import *
+
+# Cargar el modelo entrenado
+learn = load_learner('model.pkl')
+
+# Función para hacer predicciones
+def predict(image):
+    pred, pred_idx, probs = learn.predict(image)
+    return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(learn.dls.vocab))}
+
+# Crear la interfaz con Gradio
+interface = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),  # Permite cargar una imagen
+    outputs=gr.Label(),  # Muestra las predicciones con probabilidades
+    title="Clasificador de Tabaco",
+    description="Sube una imagen y el modelo la clasificará en una categoría específica."
+)
+
+# Ejecutar la aplicación
+if __name__ == "__main__":
+    interface.launch()