fix: default model for image generation

2024-05-02 08:48:58 +02:00 · 2024-05-02 08:48:58 +02:00 · 46c8d64dc5
commit 46c8d64dc5
parent a264e91f1f
1 changed files with 2 additions and 1 deletions
--- 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