Update app.py

app.py

@@ -29,8 +29,8 @@ PREVIEW_IMAGES = True
 dtype = torch.bfloat16
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
-    prior_pipeline = StableCascadePriorPipeline.from_pretrained("diffusers/StableCascade-prior", torch_dtype=dtype).to(device)
-    decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("diffusers/StableCascade-decoder",  torch_dtype=dtype).to(device) 
+    prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/StableCascade-prior", torch_dtype=dtype).to(device)
+    decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/StableCascade",  torch_dtype=dtype).to(device) 
 
     if ENABLE_CPU_OFFLOAD:
         prior_pipeline.enable_model_cpu_offload()