Update app.py

app.py

@@ -4,13 +4,16 @@ import torch
 
 # Используем официальную модель
 model_id = "CompVis/stable-diffusion-v1-4"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+
+# Загружаем модель без указания torch_dtype для работы на CPU
+pipe = StableDiffusionPipeline.from_pretrained(model_id)
+pipe.to("cpu")  # Переносим модель на CPU
 
 def generate_image(prompt):
     image = pipe(prompt).images[0]
     return image
 
+# Запуск Gradio интерфейса
 gr.Interface(
     fn=generate_image, 
     inputs="text",