Create main.py

main.py

@@ -0,0 +1,10 @@
+from prediction import predict_class
+import tensorflow as tf
+
+# Cargar el modelo
+new_model = tf.keras.models.load_model('fruits_detection.keras')
+
+# Función para manejar la solicitud de predicción
+def predict_image(image_path):
+    predicted_class = predict_class(new_model, image_path)
+    return {"predicted_class": predicted_class}