Update app.py

app.py

@@ -39,7 +39,10 @@ def contrast_loss(image_tensor):
 
 # Setup Stable Diffusion Pipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4").to(device)
+pipe = StableDiffusionPipeline.from_pretrained(
+    "CompVis/stable-diffusion-v1-4",
+    torch_dtype=torch.float16
+).to(device)
 
 # Image transform to tensor
 transform = transforms.ToTensor()