Update app.py

app.py

@@ -3,8 +3,8 @@ import spaces
 import gradio as gr
 from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
 
-prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior")#, torch_dtype=torch.float16)
-decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade")#, torch_dtype=torch.float16)
+prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.float16).to("CUDA")
+decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to("CUDA")
 
 @spaces.GPU(duration=480)
 def generate(prompt, negative_prompt, width, height, steps):