Update app.py

app.py

@@ -6,7 +6,8 @@ from huggingface_hub import hf_hub_download
 
 # Cargar el modelo desde Hugging Face Model Hub
 def load_model():
-    model_path = hf_hub_download(repo_id="tu_usuario/tu_repositorio", filename="modelo_frutas_transfer.keras")
+    # Aquí descargamos el modelo desde Hugging Face
+    model_path = hf_hub_download(repo_id="imanolcb/basicFruitClassifier", filename="modelo_frutas_transfer.keras")
     model = tf.keras.models.load_model(model_path)
     return model
 
@@ -38,8 +39,8 @@ def predict_image(image_input):
 # Interfaz Gradio
 iface = gr.Interface(
     fn=predict_image, 
-    inputs=gr.inputs.Image(type="file", label="Cargar imagen de fruta"), 
-    outputs=gr.outputs.Textbox(label="Predicción de la clase"), 
+    inputs=gr.Image(type="file", label="Cargar imagen de fruta"), 
+    outputs=gr.Textbox(label="Predicción de la clase"), 
     title="Clasificador de Frutas",
     description="Sube una imagen de una fruta y el modelo predecirá qué fruta es."
 )