Update app.py

app.py

@@ -1,37 +1,52 @@
 import torch
-from diffusers import StableDiffusionXLPipeline, AutoencoderKL
+import spaces
 import gradio as gr
+from diffusers import StableDiffusionXLPipeline, AutoencoderKL
 
-# Загружаем базовую модель Pony
+# 1. Загружаем модель один раз при старте (вне функции генерации)
 model_id = "LyliaEngine/Pony_Diffusion_V6_XL"
-vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch_float16)
+# Специальный VAE для SDXL, чтобы не было "черных квадратов"
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+
 pipe = StableDiffusionXLPipeline.from_pretrained(
     model_id, 
     vae=vae,
     torch_dtype=torch.float16, 
     variant="fp16"
-).to("cuda")
-
-# Загружаем LoRA (стиль Summertime Saga)
+)
+# Подгружаем LoRA
 pipe.load_lora_weights("nncyberpunk/LoRA_SDXL1.0_SummertimeSagaXL_Pony", weight_name="SummertimeSagaXL.safetensors")
+pipe.to("cuda") # Это сработает внутри Space
 
-def infer(prompt, negative_prompt, lora_scale):
-    # Применяем вес LoRA (обычно 0.6 - 0.8)
+# 2. Функция генерации с декоратором для ZeroGPU
+@spaces.GPU
+def infer(prompt, lora_scale):
+    # У Pony свои правила промптов, добавляем их автоматически для качества
+    base_negative = "score_6, score_5, score_4, low quality, bad anatomy, blur"
+    
     image = pipe(
         prompt=prompt,
-        negative_prompt=negative_prompt,
+        negative_prompt=base_negative,
         cross_attention_kwargs={"scale": lora_scale},
         num_inference_steps=25,
         guidance_scale=7.0,
     ).images[0]
     return image
 
-# Интерфейс как в твоем проекте Punch the Monkey
+# 3. Интерфейс Gradio
 with gr.Blocks() as demo:
-    prompt = gr.Textbox(label="Prompt (используй score_9, score_8 для Pony)")
-    lora_scale = gr.Slider(0, 1, value=0.7, label="LoRA Strength")
-    btn = gr.Button("Generate")
-    output = gr.Image()
+    gr.Markdown("# Summertime Saga XL Generator")
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(
+                label="Prompt", 
+                placeholder="score_9, score_8, rating_explicit, summertime saga style, girl in the park..."
+            )
+            lora_scale = gr.Slider(0, 1, value=0.7, label="LoRA Strength")
+            btn = gr.Button("Generate")
+        with gr.Column():
+            output = gr.Image()
+    
     btn.click(infer, inputs=[prompt, lora_scale], outputs=output)
 
 demo.launch()