Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from diffusers import DiffusionPipeline, StableDiffusionXLImg2ImgPipeline
+from diffusers import DiffusionPipeline
 import torch
 from PIL import Image
 
@@ -8,13 +8,13 @@ prj_path = "jkcg/furniture-chair"
 model = "stabilityai/stable-diffusion-xl-base-1.0"
 pipe = DiffusionPipeline.from_pretrained(
      model,
-     torch_dtype=torch.float16,
+     torch_dtype=torch.float32,  # Use float32 for CPU
 )
-pipe.to("cuda")
+pipe.to("cpu")  # Ensure the pipeline runs on CPU
 pipe.load_lora_weights(prj_path, weight_name="pytorch_lora_weights.safetensors")
 
 def generate_image(prompt, seed):
-    generator = torch.Generator("cuda").manual_seed(seed)
+    generator = torch.Generator("cpu").manual_seed(seed)
     image = pipe(prompt=prompt, generator=generator).images[0]
     return image