File size: 457 Bytes
0ebb335 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from flask import Flask, request, jsonify
from handler import EndpointHandler
app = Flask(__name__)
model_handler = EndpointHandler("./")
@app.route("/predict", methods=["POST"])
def predict():
input_data = request.json
text = input_data["text"]
# Usar el handler para hacer una predicción
prediction = model_handler.call(text)
return jsonify({"prediction": prediction})
if __name__ == "__main__":
app.run(debug=True)
|