dlflannery commited on
Commit
4411fd0
·
verified ·
1 Parent(s): ff0ca02

Update app.py

Browse files

catch openai image request returns empty list

Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -452,6 +452,9 @@ def make_image(prompt, user, pwd):
452
  except Exception as ex:
453
  msg = ex.message
454
  return [gr.Image(visible=False, value=None), msg]
 
 
 
455
  try:
456
  image_data = response.data[0].b64_json
457
  with Image.open(BytesIO(base64.b64decode(image_data))) as image:
 
452
  except Exception as ex:
453
  msg = ex.message
454
  return [gr.Image(visible=False, value=None), msg]
455
+ if len(response.data) == 0:
456
+ msg = "OpenAI returned no image data"
457
+ return [gr.Image(visible=False, value=None), msg]
458
  try:
459
  image_data = response.data[0].b64_json
460
  with Image.open(BytesIO(base64.b64decode(image_data))) as image: