Update app.py

app.py

@@ -130,7 +130,7 @@ def main():
                     
             elif selected_model == "stabilityai/stable-diffusion-xl-base-1.0" or user_input == '/image:':
                 pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
-                pipe.to("cuda")
+                pipe.to("cpu")
 
                 params = {'num_inference_steps': 100, 'num_images_per_prompt': 2}
                 generate_image(pipe, prompt, params)