Update app.py

app.py

@@ -1,17 +1,21 @@
-# Codigo de la aplicación
+from huggingface_hub import from_pretrained_fastai
 import gradio as gr
+from fastai.vision.all import *
 from fastai.learner import load_learner
 from PIL import Image
 
 # Load the learner
-learn_inf = load_learner("LuisCe/Practica03")
+repo_id ="LuisCe/Practica03"
+
+learner = from_pretrained_fastai(repo_id)
+# learn_inf = load_learner("LuisCe/Practica03")
 
 # Define the prediction function
 def predict_image(img):
     # Convert the PIL image to a format that fastai expects
-    img_fastai = Image.fromarray(img.astype('uint8'), 'RGB')
+    # img_fastai = Image.fromarray(img.astype('uint8'), 'RGB')
     # Make prediction
-    pred, _, _ = learn_inf.predict(img_fastai)
+    pred, _, _ = learner.predict(img_fastai)
     # Return prediction
     return pred