dlflannery commited on
Commit
a509fbe
·
verified ·
1 Parent(s): 4c6a845

Update app.py

Browse files

hd 1k x 1k dall-e-3 image gen with 'hd' prompt prepend

Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -429,12 +429,20 @@ def make_image(prompt, user, pwd):
429
  user = user.lower().strip()
430
  msg = 'Error: unable to create image.'
431
  fpath = None
 
 
 
432
  if user in unames and pwd == pwdList[unames.index(user)]:
433
  if len(prompt.strip()) == 0:
434
  return [gr.Image(value=None, visible=False), 'You must provide a prompt describing image you desire']
 
 
 
 
 
435
  try:
436
- response = Client().images.generate(model='dall-e-2', prompt=prompt,size='512x512',
437
- quality='standard', response_format='b64_json')
438
  except Exception as ex:
439
  msg = ex.message
440
  return [gr.Image(visible=False, value=None), msg]
 
429
  user = user.lower().strip()
430
  msg = 'Error: unable to create image.'
431
  fpath = None
432
+ model = 'dall-e-2'
433
+ size = '512x512'
434
+ quality = 'standard'
435
  if user in unames and pwd == pwdList[unames.index(user)]:
436
  if len(prompt.strip()) == 0:
437
  return [gr.Image(value=None, visible=False), 'You must provide a prompt describing image you desire']
438
+ if prompt.startswith('hd '):
439
+ prompt = prompt[3:]
440
+ model = 'dall-e-3'
441
+ size = '1024x1024'
442
+ quality = 'hd'
443
  try:
444
+ response = Client().images.generate(model=model, prompt=prompt,size=size,
445
+ quality=quality, response_format='b64_json')
446
  except Exception as ex:
447
  msg = ex.message
448
  return [gr.Image(visible=False, value=None), msg]