fix: fix model names

2024-05-02 08:57:50 +02:00
parent 46c8d64dc5
commit 24ee69b98e
2 changed files with 7 additions and 6 deletions
--- a/gen_service/main.py
+++ b/gen_service/main.py
@ -108,9 +108,10 @@ def _gen_text(prompt, use_zukijourney=False):

@app.route('/gen_image', methods=['GET'])
 def gen_image():
+    print(request.args)
    prompt = request.args.get('prompt')
    model = request.args.get('model')
-    model = model if model else "playgroind-v2.5"
+    model = model if model else "playground-v2.5"

    if not prompt:
        return "Error: No prompt provided", 400