Update app.py

app.py

@@ -1,4 +1,5 @@
-
+# Шаг 1. Установка
+!pip install diffusers transformers accelerate safetensors gradio --quiet
 
 # Шаг 2. Импорт
 import torch
@@ -7,29 +8,48 @@ from PIL import Image, ImageFilter, ImageEnhance
 import numpy as np
 import gradio as gr
 
-# Шаг 3. Загрузка модели
+# Шаг 3. Проверка GPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Используется: {device}")
+
+# Шаг 4. Загрузка модели (без float16 для CPU)
 print("Загрузка модели...")
 model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16,
-    safety_checker=None,
-    requires_safety_checker=False
-)
-pipe = pipe.to("cpu")
-pipe.enable_attention_slicing()
-pipe.enable_vae_slicing()
+
+if device == "cuda":
+    pipe = StableDiffusionPipeline.from_pretrained(
+        model_id,
+        torch_dtype=torch.float16,
+        safety_checker=None,
+        requires_safety_checker=False
+    )
+else:
+    pipe = StableDiffusionPipeline.from_pretrained(
+        model_id,
+        safety_checker=None,
+        requires_safety_checker=False
+    )
+
+pipe = pipe.to(device)
+
+# Экономия памяти
+try:
+    pipe.enable_attention_slicing()
+except:
+    pass
+
+try:
+    pipe.vae.enable_slicing()
+except:
+    pass
+
 print("Модель загружена!")
 
-# Шаг 4. Функция генерации
+# Шаг 5. Функция генерации
 def generate_image(prompt, negative_prompt):
-    # УСИЛЕНИЕ ПРОМПТА (если он короткий)
-    if len(prompt.split()) < 10:
-        prompt = prompt + ", detailed, sharp focus, masterpiece"
-    
-    generator = torch.Generator(device="cuda").manual_seed(2021)
+    generator = torch.Generator(device=device).manual_seed(2021)
     
-    with torch.autocast("cuda"):
+    with torch.no_grad():
         image = pipe(
             prompt=prompt,
             negative_prompt=negative_prompt if negative_prompt else "",
@@ -56,7 +76,7 @@ def generate_image(prompt, negative_prompt):
     
     return img_final
 
-# Шаг 5. Интерфейс
+# Шаг 6. Интерфейс
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("#")
     
@@ -83,5 +103,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=output
     )
 
-# Шаг 6. Запуск
+# Шаг 7. Запуск
 demo.launch(share=True, debug=False)