Update app.py

app.py

@@ -7,7 +7,7 @@ from diffusers import DiffusionPipeline, StableDiffusionLatentUpscalePipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4", safety_checker=None)
-upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16, safety_checker=None)
+upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", safety_checker=None)
 upscaler = upscaler.to(device)
 pipe = pipe.to(device)