Update app.py

app.py

@@ -1,17 +1,13 @@
 import gradio as gr
+from fastai.vision.all import *
 from PIL import Image
-import torch
-from transformers import AutoModelForImageClassification, AutoFeatureExtractor
 
-model = AutoModelForImageClassification.from_pretrained("AdrianRevi/Practica1Blindness")
-extractor = AutoFeatureExtractor.from_pretrained("AdrianRevi/Practica1Blindness")
+# Cargar modelo exportado desde Hugging Face
+learn = load_learner("https://huggingface.co/AdrianRevi/Practica1Blindness/model.pkl")
 
 def predict(img: Image.Image):
-    inputs = extractor(images=img, return_tensors="pt")
-    with torch.no_grad():
-        outputs = model(**inputs)
-        probs = torch.nn.functional.softmax(outputs.logits, dim=1)[0]
-    labels = model.config.id2label
+    pred_class, pred_idx, probs = learn.predict(img)
+    labels = learn.dls.vocab
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 demo = gr.Interface(