Update app.py

app.py

@@ -2,12 +2,10 @@ import os
 import gradio as gr
 import numpy as np
 from gradio_client import Client
-from PIL import Image
 
-# Используем переменные окружения для гибкости
 MODEL_ID = os.getenv("MODEL_ID", "KingNish/SDXL-Flash")
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
-BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))  # Позволяет генерировать несколько изображений за один раз
+BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
 
 client = Client(MODEL_ID)
 
@@ -138,11 +136,9 @@ with gr.Blocks(css=css) as demo:
                 use_resolution_binning=True,
                 api_name="/run"
             )
-            # Проверяем, если ответ корректный, то добавляем изображение в список
-            if isinstance(response, list) and "image" in response[0]:
+            if isinstance(response, list) and response[0].get("image"):
                 results.append(response[0]["image"])
             else:
-                # В случае некорректного ответа добавляем пустое изображение
                 results.append("")
         return results, seed