Spaces:
Running
Running
Update app.py
Browse filescatch openai image request returns empty list
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:
|