Update app.py

app.py

@@ -2,9 +2,14 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+# Выбираем модель, которая точно поддерживает CPU
+model_id = "runwayml/stable-diffusion-v1-5" 
+
+# Отключаем использование высокопроизводительных инструкций xFormers, так как они не нужны на CPU
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, use_auth_token=True, variant="fp16", revision="fp16")
+
+# Не перемещаем модель на GPU
+# pipe = pipe.to("cuda") 
 
 def generate_image(prompt, guidance_scale=7.5, num_inference_steps=50):
   image = pipe(prompt, guidance_scale=guidance_scale, num_inference_steps=num_inference_steps).images[0]