From 24ee69b98ef45799aca6759f78415a8d65eea4de Mon Sep 17 00:00:00 2001 From: Yandrik Date: Thu, 2 May 2024 08:57:50 +0200 Subject: [PATCH] fix: fix model names --- gen_service/main.py | 3 ++- web/src/routes/+page.svelte | 10 +++++----- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/gen_service/main.py b/gen_service/main.py index f1c7698..15a9898 100644 --- 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 diff --git a/web/src/routes/+page.svelte b/web/src/routes/+page.svelte index 2a7f832..7826c0c 100644 --- a/web/src/routes/+page.svelte +++ b/web/src/routes/+page.svelte @@ -6,11 +6,11 @@ const models = [ { - name: "Runway v2.5", - value: "runway-v2.5", + name: "Playground v2.5", + value: "playground-v2.5", }, { - name: "Dall-e 3", + name: "Dall-E 3", value: "dall-e-3", }, { @@ -18,8 +18,8 @@ value: "stable-diffusion-3", }, { - name: "Pollinations", - value: "pollinations" + name: "DreamShaper XL Turbo", + value: "dreamshaper-xl-turbo" }, { name: "RealVis XL 4",