Yandrik/lottis_imagen
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: fix model names

Browse Source
This commit is contained in:
Yandrik 2024-05-02 08:57:50 +02:00
parent 46c8d64dc5
commit 24ee69b98e
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gen_service/main.py
View File

@ -108,9 +108,10 @@ def _gen_text(prompt, use_zukijourney=False):
@app.route('/gen_image', methods=['GET']) @app.route('/gen_image', methods=['GET'])
def gen_image(): def gen_image():
print(request.args)
prompt = request.args.get('prompt') prompt = request.args.get('prompt')
model = request.args.get('model') model = request.args.get('model')
model = model if model else "playgroind-v2.5" model = model if model else "playground-v2.5"
if not prompt: if not prompt:
return "Error: No prompt provided", 400 return "Error: No prompt provided", 400

10
web/src/routes/+page.svelte
View File

@ -6,11 +6,11 @@
const models = [ const models = [
{ {
name: "Runway v2.5", name: "Playground v2.5",
value: "runway-v2.5", value: "playground-v2.5",
}, },
{ {
name: "Dall-e 3", name: "Dall-E 3",
value: "dall-e-3", value: "dall-e-3",
}, },
{ {
@ -18,8 +18,8 @@
value: "stable-diffusion-3", value: "stable-diffusion-3",
}, },
{ {
name: "Pollinations", name: "DreamShaper XL Turbo",
value: "pollinations" value: "dreamshaper-xl-turbo"
}, },
{ {
name: "RealVis XL 4", name: "RealVis XL 4",