Rooni commited on
Commit
bfa67c7
·
verified ·
1 Parent(s): 59953d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -138,7 +138,12 @@ with gr.Blocks(css=css) as demo:
138
  use_resolution_binning=True,
139
  api_name="/run"
140
  )
141
- results.append(response[0]["image"]) # Возвращаем только изображения
 
 
 
 
 
142
  return results, seed
143
 
144
  gr.on(
 
138
  use_resolution_binning=True,
139
  api_name="/run"
140
  )
141
+ # Проверяем, если ответ корректный, то добавляем изображение в список
142
+ if isinstance(response, list) and "image" in response[0]:
143
+ results.append(response[0]["image"])
144
+ else:
145
+ # В случае некорректного ответа добавляем пустое изображение
146
+ results.append("")
147
  return results, seed
148
 
149
  gr.on(