fix : none generator error
This commit is contained in:
+3
-1
@@ -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)
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user