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

fix: default model for image generation

Browse Source
This commit is contained in:
Yandrik 2024-05-02 08:48:58 +02:00
parent a264e91f1f
commit 46c8d64dc5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gen_service/main.py
View File

@ -109,7 +109,8 @@ 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():
prompt = request.args.get('prompt') 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: if not prompt:
return "Error: No prompt provided", 400 return "Error: No prompt provided", 400