Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 from io import BytesIO
 from diffusers import StableDiffusionImg2ImgPipeline
 
-device="cuda"
+device="cpu"
 
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=st.secrets['USER_TOKEN'])
 pipe.to(device)
@@ -17,7 +17,7 @@ def resize(value,img):
   return img
 
 def infer(source_img, prompt, guide, steps, seed, Strength): 
-    generator = torch.Generator('cuda').manual_seed(seed)     
+    generator = torch.Generator('cpu').manual_seed(seed)     
     source_image = resize(512, source_img)
     source_image.save('source.png')
     image_list = pipe([prompt], init_image=source_image, strength=Strength, guidance_scale=guide, num_inference_steps=steps)