Update app.py

app.py

@@ -55,7 +55,7 @@ class StableDiffusionLoader:
             self.pretrain_pipe, 
             #torch_dtype=torch.float16, 
             use_auth_token=False
-            ).to(device)
+            ).to(self.device)
         pipe = pipe.to(self.device)
         with autocast(self.device):
             image = pipe(self.prompt)[0][0]