diff --git a/gen_service/main.py b/gen_service/main.py index c5e83a1..f1c7698 100644 --- a/gen_service/main.py +++ b/gen_service/main.py @@ -109,7 +109,8 @@ def _gen_text(prompt, use_zukijourney=False): @app.route('/gen_image', methods=['GET']) def gen_image(): prompt = request.args.get('prompt') - model = request.args.get('model', 'playground-v2.5') + model = request.args.get('model') + model = model if model else "playgroind-v2.5" if not prompt: return "Error: No prompt provided", 400