nico9ga/fruits-Classification-Model

1from prediction import predict_class2import tensorflow as tf3 4# Cargar el modelo5new_model = tf.keras.models.load_model('fruits_detection.keras')6 7# Función para manejar la solicitud de predicción8def predict_image(image_path):9    predicted_class = predict_class(new_model, image_path)10    return {"predicted_class": predicted_class}