Update app.py

app.py

@@ -20,35 +20,36 @@ pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-@spaces.GPU(duration=45)
+def calculate_duration(width, height, num_inference_steps):
+    # Примерная оценка времени выполнения
+    base_time = 10  # Минимальное время для 512x512, 10 шагов
+    time_factor = (width * height) / (512 * 512)  # Увеличение времени от размера
+    step_factor = num_inference_steps / 10  # Увеличение времени от числа шагов
+    return int(base_time * time_factor * step_factor)
+examples = [
+    "A border collie lying in some Fall leaves as the forest trees change colors",
+]
+@spaces.GPU(duration=30)  # Устанавливаем 30 секунд
 def infer(
     prompt,
-    negative_prompt="",
-    seed=42,
-    randomize_seed=False,
-    width=768,
-    height=1024,
-    guidance_scale=4.5,
-    num_inference_steps=20,
-    progress=gr.Progress(track_tqdm=True),
+    width=512,
+    height=512,
+    num_inference_steps=20,  # Оптимальные шаги
+    guidance_scale=7.5,
 ):
-    try:
-        if randomize_seed:
-            seed = random.randint(0, MAX_SEED)
-
-        generator = torch.Generator().manual_seed(seed)
-
-        image = pipe(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
-            guidance_scale=guidance_scale,
-            num_inference_steps=num_inference_steps,
-            width=width,
-            height=height,
-            generator=generator,
-        ).images[0]
+    # Оценка необходимого времени
+    duration = calculate_duration(width, height, num_inference_steps)
+    print(f"Запрашиваемая квота: {duration} секунд")
+    # Настраиваем параметры в pipe
+    image = pipe(
+        prompt=prompt,
+        num_inference_steps=num_inference_steps,
+        width=width,
+        height=height,
+        guidance_scale=guidance_scale,
+    ).images[0]
+    return image
 
-        return image, seed
 
     except Exception as e:
         # Log the exception if needed
@@ -60,9 +61,7 @@ def infer(
             "Please wait a moment and try again."
         )
 
-examples = [
-    "A border collie lying in some Fall leaves as the forest trees change colors",
-]
+
 
 # Define the custom theme with input styles
 class CustomTheme(gr.themes.Base):