diff --git a/gen_service/main.py b/gen_service/main.py index d42ae9b..c5e83a1 100644 --- a/gen_service/main.py +++ b/gen_service/main.py @@ -109,6 +109,7 @@ 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') if not prompt: return "Error: No prompt provided", 400 @@ -119,7 +120,7 @@ def gen_image(): try: client = get_client() response = client.images.generate( - model="playground-v2.5", + model=model, prompt=prompt, ) image_url = response.data[0].url diff --git a/web/src/routes/+page.svelte b/web/src/routes/+page.svelte index 86a1fa4..2a7f832 100644 --- a/web/src/routes/+page.svelte +++ b/web/src/routes/+page.svelte @@ -1,9 +1,38 @@