Update app.py

app.py

@@ -12,7 +12,7 @@ auth_token = os.environ.get("auth_token") #in secret space
 
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=auth_token)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-pipe.to(device)
+pipe = pipe.to(device)
 
 def encode_pil_to_base64(pil_image):
   ''' From: https://github.com/gradio-app/gradio/blob/main/gradio/processing_utils.py'''