commit

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Лёгкая модель — работает на бесплатном GPU/CPU
-pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", torch_dtype=torch.float16, safety_checker=None)
+# Публичная модель без gated (работает без token)
+pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, safety_checker=None)
 pipe.enable_model_cpu_offload()
 pipe.enable_vae_slicing()
 pipe.enable_attention_slicing()
@@ -16,14 +16,14 @@ def generate(prompt, negative_prompt="", steps=30, seed=42):
         image = pipe(prompt, negative_prompt=negative, num_inference_steps=steps, generator=generator).images[0]
         return image
     except Exception as e:
-        return f"Ошибка: {str(e)}. Попробуй проще prompt или подожди."
+        return f"Ошибка: {str(e)}. Попробуй проще prompt."
 
 with gr.Blocks(theme=gr.themes.Dark()) as demo:
     gr.Markdown("# Epic AI Art Generator 🔥")
-    gr.Markdown("Сделано для моей девушки ❤️ Попробуй на русском или английском!")
+    gr.Markdown("Сделано для моей девушки ❤️ Генерирует эпичные арты! Попробуй на русском или английском.")
 
     with gr.Row():
-        prompt = gr.Textbox(label="Prompt (описание)", placeholder="рыжий кот в стиле Гарри Поттера")
+        prompt = gr.Textbox(label="Prompt (описание)", placeholder="рыжий кот в стиле Гарри Поттера, волшебная шляпа")
         negative = gr.Textbox(label="Negative prompt", placeholder="blurry, low quality")
         steps = gr.Slider(20, 50, value=30, step=5, label="Шаги (steps)")
         seed = gr.Slider(0, 999999, value=42, label="Seed")