app.py

app.py

@@ -2,23 +2,27 @@ import gradio as gr
 from diffusers import StableDiffusionXLPipeline
 import torch
 
-# Load Turbo model with optimizations
+# Load Turbo model with max optimization
 pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", safety_checker=None)
-pipe.enable_model_cpu_offload()  # Ключ для бесплатного tier
+pipe.enable_model_cpu_offload()  # Разгружает на CPU, если GPU мало
 pipe.enable_vae_slicing()
+pipe.enable_attention_slicing()
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
 def generate(prompt, steps=4, guidance=0.0, seed=42):
-    generator = torch.Generator("cuda").manual_seed(seed) if torch.cuda.is_available() else None
-    image = pipe(prompt, num_inference_steps=steps, guidance_scale=guidance, generator=generator).images[0]
-    return image
+    try:
+        generator = torch.Generator("cuda").manual_seed(seed) if torch.cuda.is_available() else None
+        image = pipe(prompt, num_inference_steps=steps, guidance_scale=guidance, generator=generator).images[0]
+        return image
+    except Exception as e:
+        return f"Ошибка генерации: {str(e)}. Попробуй проще prompt или меньше шагов. GPU перегружен."
 
 with gr.Blocks() as demo:
     gr.Markdown("# Epic AI Art Generator 🔥")
     gr.Markdown("Сделано для моей девушки ❤️ Генерирует арты за секунды! Попробуй на русском или английском.")
 
     with gr.Row():
-        prompt = gr.Textbox(label="Prompt (описание)", placeholder="рыжий кот космонавт или romantic couple at sunset")
+        prompt = gr.Textbox(label="Prompt (описание)", placeholder="кот в стиле Гарри Поттера")
         steps = gr.Slider(1, 10, value=4, step=1, label="Шаги (steps) — 4 для скорости")
         guidance = gr.Slider(0, 5, value=0.0, step=0.5, label="Guidance (0 для Turbo)")
         seed = gr.Slider(0, 999999, value=42, label="Seed")
@@ -32,6 +36,6 @@ with gr.Blocks() as demo:
     gr.Markdown("### Примеры:")
     gr.Markdown("- 'милый рыжий кот в космосе'")
     gr.Markdown("- 'романтическая пара на пляже на закате'")
-    gr.Markdown("- 'супергеройская пара в городе'")
+    gr.Markdown("- 'кот в стиле Гарри Поттера'")
 
 demo.launch()