Omnibus commited on
Commit
eb89384
1 Parent(s): ada9a4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -425,8 +425,8 @@ def get_prompts(prompt_text):
425
  gen=text_gen(prompt_text)
426
  return gen
427
 
428
- def gen_im(prompt,model_choice,count=1):
429
- out = im_gen(prompt,int(model_choice),count)
430
  return out
431
 
432
 
 
425
  gen=text_gen(prompt_text)
426
  return gen
427
 
428
+ def gen_im(prompt,model_choice=5,count=1):
429
+ out = im_gen(prompt,5,count)
430
  return out
431
 
432