Create app.py

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from fastai.vision.all import *
+
+learn = load_learner("model.pkl")
+
+categorias = learn.dls.vocab
+
+def predict(img):
+    img = PILImage.create(img)
+    pred, pred_idx, probs = learn.predict(img)
+    return dict(zip(categorias, map(float, probs)))
+
+# Interfaz de Gradio
+titulo = "Clasificador de Vehículos - Entregable 2"
+descripcion = "Detector de vehículos (Bikes, Cars, Cabs, etc.) entrenado con FastAI."
+
+interface = gr.Interface(
+    fn=predict, 
+    inputs=gr.Image(), 
+    outputs=gr.Label(num_top_classes=3),
+    title=titulo,
+    description=descripcion,
+    examples=["Bike (104).jpg", "Car (100).jpg"] 
+)
+
+interface.launch()