From 46c8d64dc572e1973382d40fe1ebc56e0ebb5660 Mon Sep 17 00:00:00 2001 From: Yandrik Date: Thu, 2 May 2024 08:48:58 +0200 Subject: [PATCH] fix: default model for image generation --- gen_service/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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