update app.py

app.py

@@ -2,19 +2,66 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 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
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-def generate(prompt):
-    image = pipe(prompt, num_inference_steps=30).images[0]  # 30 шагов для скорости
-    return image
+# 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"
+
+    # Seed for reproducibility
+    generator = torch.Generator("cuda").manual_seed(seed) if torch.cuda.is_available() else None
+
+    # 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.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 (для повторяемости)")
+
+            generate_btn = gr.Button("Generate Magic 🔥", variant="primary")
+
+        with gr.Column(scale=1):
+            output_image = gr.Image(label="Твоё искусство")
+
+            gr.Markdown("### Примеры промптов для нас:")
+            gr.Markdown("- 'романтическая пара на пляже на закате, реалистично'")
+            gr.Markdown("- 'мы как супергерои в городе'")
+            gr.Markdown("- 'рыжий кот в фэнтези замке'")
 
-demo = gr.Interface(
-    fn=generate,
-    inputs=gr.Textbox(label="Введите описание (prompt, можно на русском)"),
-    outputs=gr.Image(label="Сгенерированное изображение"),
-    title="Epic AI Art Generator 🔥",
-    description="Генерирует арты по тексту. Попробуй: 'милый кот в космосе' или 'романтическая пара на закате' ❤️"
-)
+    # Action
+    generate_btn.click(
+        fn=generate_image,
+        inputs=[prompt_input, negative_input, steps_slider, seed_slider],
+        outputs=output_image
+    )
 
-demo.launch()
+demo.launch(debug=True)  # Debug для логов, если что