update app.py

app.py

@@ -1,67 +1,38 @@
 import gradio as gr
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionXLPipeline
 import torch
 
-# Load optimized model (float16 for speed and memory — from your efficiency course)
-pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, safety_checker=None)
-
-# Optimizations for free GPU (key to avoid long processing)
-pipe.enable_vae_slicing()
-pipe.enable_attention_slicing()
-pipe.enable_sequential_cpu_offload()  # Crucial for low memory
+# Load SDXL-Turbo (super fast, 4–10 sec per image, great quality)
+pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Generation function
-def generate_image(prompt, negative_prompt="", num_steps=25, seed=42):
-    # Negative prompt for better quality
-    negative = negative_prompt or "blurry, low quality, ugly, deformed, bad anatomy"
+# Optimizations for free tier
+pipe.enable_model_cpu_offload()
 
-    # Seed for reproducibility
+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
 
-    # Generation (25 steps — fast and good quality)
-    result = pipe(
-        prompt,
-        negative_prompt=negative,
-        num_inference_steps=num_steps,
-        generator=generator
-    ).images[0]
-
-    return result
-
-# Gradio interface (beautiful and user-friendly)
-with gr.Blocks(title="Epic AI Art Generator 🔥") as demo:
-    gr.Markdown("# Epic AI Art Generator")
-    gr.Markdown("Made with ❤️ for my amazing girl. Generate our adventures in any style! Попробуй: 'рыжий кот в космосе' или 'романтическая пара на закате'")
+with gr.Blocks() as demo:
+    gr.Markdown("# Мой Epic AI Generator 🔥")
+    gr.Markdown("Сделано для моей девушки ❤️ Генерирует арты за секунды! Попробуй на русском или английском.")
 
     with gr.Row():
-        with gr.Column(scale=1):
-            prompt_input = gr.Textbox(
-                label="Описание (prompt на русском или английском)",
-                placeholder="e.g. 'милый рыжий кот космонавт' or 'beautiful couple at sunset beach'"
-            )
-            negative_input = gr.Textbox(
-                label="Negative prompt (что избегать)",
-                placeholder="blurry, low quality, ugly"
-            )
-            steps_slider = gr.Slider(15, 50, value=25, step=5, label="Шаги (steps) — больше = лучше качество, но дольше")
-            seed_slider = gr.Slider(0, 999999999, value=42, step=1, label="Seed (для повторяемости)")
+        prompt = gr.Textbox(label="Prompt (описание)", placeholder="рыжий кот космонавт, highly detailed")
+        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")
 
-            generate_btn = gr.Button("Generate Magic 🔥", variant="primary")
+        btn = gr.Button("Generate 🔥")
 
-        with gr.Column(scale=1):
-            output_image = gr.Image(label="Твоё искусство")
+    output = gr.Image(label="Арт")
 
-            gr.Markdown("### Примеры промптов для нас:")
-            gr.Markdown("- 'романтическая пара на пляже на закате, реалистично'")
-            gr.Markdown("- 'мы как супергерои в городе'")
-            gr.Markdown("- 'рыжий кот в фэнтези замке'")
+    btn.click(generate, inputs=[prompt, steps, guidance, seed], outputs=output)
 
-    # Action
-    generate_btn.click(
-        fn=generate_image,
-        inputs=[prompt_input, negative_input, steps_slider, seed_slider],
-        outputs=output_image
-    )
+    gr.Markdown("### Примеры:")
+    gr.Markdown("- 'милый рыжий кот в космосе'")
+    gr.Markdown("- 'романтическая пара на закате пляжа'")
+    gr.Markdown("- 'супергеройская пара в городе'")
 
-demo.launch(debug=True)  # Debug для логов, если что
+demo.launch()