Update app.py

app.py

@@ -1,6 +1,5 @@
 from fastai.vision.all import load_learner
 from huggingface_hub import hf_hub_download
-from fastai.callback.fp16 import AMPMode
 import gradio as gr
 
 # Descarga el modelo desde Hugging Face
@@ -9,6 +8,7 @@ learn = load_learner(model_path)
 
 # Función para predicción
 def predict(image):
+    # Realiza la predicción usando el modelo cargado
     pred, _, probs = learn.predict(image)
     return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(probs))}