fix : response timeout

This commit is contained in:
martin legrand
2025-03-29 14:23:53 +01:00
parent 5c9ada9468
commit 90894f806a
2 changed files with 12 additions and 1 deletions
+11
View File
@@ -36,6 +36,17 @@ def setup():
generator.set_model(model)
return jsonify({"message": "Model set"}), 200
@app.route('/get_complete_sentence', methods=['GET'])
def get_complete_sentence():
if not generator:
return jsonify({"error": "Generator not initialized"}), 400
while True:
status = generator.get_status()
print(status)
if status["is_complete"]:
return jsonify(status)
return None
@app.route('/get_updated_sentence')
def get_updated_sentence():
if not generator: