Update app.py

app.py

@@ -13,7 +13,7 @@ from gradio.components import Textbox, Image
 
 
 repo_name = 'mohansathya/twosd' # YOUR REPO NAME
-pipe2 = StableDiffusionPipeline.from_pretrained(repo_name, torch_dtype=torch.float16)
+pipe2 = StableDiffusionPipeline.from_pretrained(repo_name, torch_dtype=torch.float16).to('cpu')
 
 
 
@@ -27,7 +27,7 @@ def generate_query_response(prompt):
 
     seed = random.randint(0, 2147483647)
     print("Seed: {}".format(str(seed)))
-    generator = torch.Generator().manual_seed(seed)
+    generator = torch.Generator(device='cpu').manual_seed(seed)
 
     with autocast("cpu"), torch.inference_mode():
         imgs = pipe2(