Update app.py

app.py

@@ -7,6 +7,8 @@ from io import BytesIO
 import torch
 
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+pipe.to(device)
 
 def encode_pil_to_base64(pil_image):
   ''' From: https://github.com/gradio-app/gradio/blob/main/gradio/processing_utils.py'''