Create app.py

app.py

@@ -0,0 +1,29 @@
+import gradio as gr
+import tensorflow as tf
+from tensorflow.keras.models import load_model
+from tensorflow.keras.preprocessing import image
+import numpy as np
+
+# Cargar el modelo
+model = load_model('https://huggingface.co/imanolcb/basicFruitClassifier/resolve/main/modelo_frutas_transfer.keras')
+
+# Función para la predicción
+def predict(img):
+    img = image.load_img(img, target_size=(150, 150))
+    img_array = image.img_to_array(img) / 255.0  # Normalización
+    img_array = np.expand_dims(img_array, axis=0)  # Cambiar la forma para predicción
+    prediction = model.predict(img_array)
+    class_names = ['manzana', 'naranja', 'platano']  # Modificar con tus clases
+    predicted_class = class_names[np.argmax(prediction)]
+    return predicted_class
+
+# Interfaz de Gradio
+iface = gr.Interface(fn=predict, 
+                     inputs=gr.Image(type="filepath"), 
+                     outputs=gr.Text(),
+                     live=True, 
+                     title="Clasificador de Frutas",
+                     description="Cargar una imagen de fruta para clasificarla.")
+
+# Iniciar la interfaz
+iface.launch()