main.py

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}