dlflannery commited on
Commit
a97a8df
·
verified ·
1 Parent(s): f19128a

Update app.py

Browse files

OpenAI dall-e2 would not accept quality='standard'

Files changed (1) hide show
  1. app.py +14 -7
app.py CHANGED
@@ -905,8 +905,8 @@ def make_image(prompt, user, pwd):
905
  msg = 'Error: unable to create image.'
906
  fpath = None
907
  model = 'dall-e-2'
908
- size = '512x512'
909
  quality = 'standard'
 
910
  if user in unames and pwd == pwdList[unames.index(user)]:
911
  if len(prompt.strip()) == 0:
912
  return [gr.Image(value=None, visible=False), 'You must provide a prompt describing image you desire']
@@ -915,12 +915,19 @@ def make_image(prompt, user, pwd):
915
  model = 'dall-e-3'
916
  size = '1024x1024'
917
  quality = 'hd'
918
- try:
919
- response = Client().images.generate(model=model, prompt=prompt,size=size,
920
- quality=quality, response_format='b64_json')
921
- except Exception as ex:
922
- msg = ex.message
923
- return [gr.Image(visible=False, value=None), msg]
 
 
 
 
 
 
 
924
  if len(response.data) == 0:
925
  msg = "OpenAI returned no image data"
926
  return [gr.Image(visible=False, value=None), msg]
 
905
  msg = 'Error: unable to create image.'
906
  fpath = None
907
  model = 'dall-e-2'
 
908
  quality = 'standard'
909
+ size = '512x512'
910
  if user in unames and pwd == pwdList[unames.index(user)]:
911
  if len(prompt.strip()) == 0:
912
  return [gr.Image(value=None, visible=False), 'You must provide a prompt describing image you desire']
 
915
  model = 'dall-e-3'
916
  size = '1024x1024'
917
  quality = 'hd'
918
+ try:
919
+ response = Client().images.generate(model=model, prompt=prompt,size=size,
920
+ response_format='b64_json', quality=quality)
921
+ except Exception as ex:
922
+ msg = ex.message
923
+ return [gr.Image(visible=False, value=None), msg]
924
+ else:
925
+ try:
926
+ response = Client().images.generate(model=model, prompt=prompt,size=size,
927
+ response_format='b64_json')
928
+ except Exception as ex:
929
+ msg = ex.message
930
+ return [gr.Image(visible=False, value=None), msg]
931
  if len(response.data) == 0:
932
  msg = "OpenAI returned no image data"
933
  return [gr.Image(visible=False, value=None), msg]