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("reroti/DreamnetV1-3", torch_dtype=torch.float16, safety_checker=None)
-upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("SG161222/Realistic_Vision_V2.0", torch_dtype=torch.float16)
+upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16)
 upscaler = upscaler.to(device)
 pipe = pipe.to(device)