Spaces:
Runtime error
Runtime error
import os | |
import logging | |
from flask import Flask, request, jsonify | |
from final import Translator | |
app = Flask(_name_) | |
model_path = 'model.h5' | |
model = Translator(model_path) | |
logging.basicConfig(level=logging.INFO) | |
def index(): | |
"""Provide a simple health check route.""" | |
return "Hello world!" | |
def predict(): | |
"""Provide the main prediction API route. Responds to both GET and POST requests.""" | |
logging.info("Predict request received!") | |
question_text = request.args.get("question_text") | |
predicted_answer = model.predict(question_text) | |
logging.info("Predicted answer: {}".format(predicted_answer)) | |
return jsonify({"predicted_answer": predicted_answer}) | |
def main(): | |
"""Run the Flask app.""" | |
app.run(host="0.0.0.0", port=8000, debug=False) | |
if _name_ == "_main_": | |
main() |