Create app.py

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from fastai.vision.all import load_learner, PILImage
+
+# Carga el modelo exportado (export.pkl) que hayas subido al repositorio
+model = load_learner('export.pkl')
+
+# Función de predicción
+def predict_image(img_path):
+    img = PILImage.create(img_path)
+    pred, pred_idx, probs = model.predict(img)
+    return {str(pred): float(probs[pred_idx])}
+
+# Interfaz Gradio
+def main():
+    iface = gr.Interface(
+        fn=predict_image,
+        inputs=gr.Image(type='filepath', label='Sube una imagen de documento Tobacco'),
+        outputs=gr.Label(num_top_classes=3, label='Predicciones'),
+        title='Clasificador Tobacco-3482',
+        description='Clasifica imágenes del dataset Tobacco-3482 en sus categorías.'
+    )
+    # En Spaces, share=True no es necesario
+    iface.launch()
+
+if __name__ == '__main__':
+    main()