fix : server error

This commit is contained in:
martin legrand 2025-03-29 12:34:51 +01:00
parent 18f23db0fa
commit d1b20a1446

View File

@ -33,6 +33,7 @@ def start_generation():
@app.route('/setup', methods=['POST']) @app.route('/setup', methods=['POST'])
def setup(): def setup():
global generator
data = request.get_json() data = request.get_json()
model = data.get('model', None) model = data.get('model', None)
provider = args.provider provider = args.provider
@ -49,6 +50,7 @@ def setup():
@app.route('/get_updated_sentence') @app.route('/get_updated_sentence')
def get_updated_sentence(): def get_updated_sentence():
global generator
if not generator: if not generator:
return jsonify({"error": "Generator not initialized"}), 400 return jsonify({"error": "Generator not initialized"}), 400
return generator.get_status() return generator.get_status()