fix : none generator error

This commit is contained in:
martin legrand
2025-03-29 12:29:38 +01:00
parent 8106cff45f
commit 18f23db0fa
3 changed files with 3 additions and 3 deletions
+3 -1
View File
@@ -49,7 +49,9 @@ def setup():
@app.route('/get_updated_sentence')
def get_updated_sentence():
return jsonify(generator.get_status())
if not generator:
return jsonify({"error": "Generator not initialized"}), 400
return generator.get_status()
if __name__ == '__main__':
app.run(host='0.0.0.0', threaded=True, debug=True, port=args.port)
-1
View File
@@ -41,7 +41,6 @@ class GeneratorLLM():
def get_status(self) -> dict:
with self.state.lock:
print("Returning status:\n", self.state.status())
return jsonify(self.state.status())
@abstractmethod