Update stable_cascade.py

stable_cascade.py

@@ -1,12 +1,12 @@
 import torch, os
 from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
 import gradio as gr
-
-os.environ['TOKENIZERS_PARALLELISM'] = 'false'
+import spaces
 
 prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16).to("cuda")
 decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to("cuda")
 
+@spaces.GPU
 def generate_images(
     prompt="a photo of a girl",
     negative_prompt="bad,ugly,deformed",