Update app.py

app.py

@@ -33,16 +33,18 @@ pipe_hyper = DiffusionPipeline.from_pretrained(base_model_id, unet=unet, torch_d
 pipe_hyper.scheduler = LCMScheduler.from_config(pipe_hyper.scheduler.config)
 pipe_hyper.to("cuda")
 
+
+@spaces.GPU
 def run(prompt):
     image_turbo=pipe_turbo(prompt=prompt, num_inference_steps=1, guidance_scale=0).images[0]
     image_lightning=pipe_lightning(prompt=prompt, num_inference_steps=1, guidance_scale=0).images[0]
     image_hyper=pipe_hyper(prompt=prompt, num_inference_steps=1, guidance_scale=0, timesteps=[800]).images[0]
     return image_turbo, image_lightning, image_hyper
+
+
 css = '''
 .gradio-container{max-width: 768px !important}
 '''
-
-@spaces.GPU
 with gr.Blocks(css=css) as demo:
     prompt = gr.Textbox(label="Prompt")
     run = gr.Button("Run")