Update app.py

app.py

@@ -14,6 +14,7 @@ from torchvision.transforms import functional as F
 import tempfile
 from diffusers import DiffusionPipeline
 import random
+import time
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -37,7 +38,7 @@ pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-# Функция для генерации изображения с помощью SDXL-Turbo
+# Функция для генерации изображения с задержкой
 def generate_with_diffusion(
     prompt,
     negative_prompt="",
@@ -47,12 +48,20 @@ def generate_with_diffusion(
     height=1024,
     guidance_scale=0.0,
     num_inference_steps=2,
+    progress=gr.Progress(),
 ):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
     generator = torch.Generator().manual_seed(seed)
 
+    # Обратный отсчет 20 секунд
+    for i in range(20, 0, -1):
+        progress((20 - i) / 20, desc=f"Генерация начнется через {i} секунд...")
+        time.sleep(20)
+
+    # Генерация изображения
+    progress(1.0, desc="Генерация изображения...")
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,