import tensorflow as tf
from tensorflow.keras.models import load_model
import numpy as np
from flask import Flask, request, jsonify

app = Flask(__name__)

# Cargar el modelo guardado
model = load_model('my_model.h5')

@app.route('/predict', methods=['POST'])
def predict():
    data = request.get_json(force=True)
    input_data = np.array([data['input']])
    prediction = model.predict(input_data)
    output = prediction[0][0]
    return jsonify({'prediction': output})

if __name__ == '__main__':
    # Ejecutar la aplicación Flask en Hugging Face con puerto 80
    app.run(host='0.0.0.0', port=80)