Update app.py

app.py

@@ -20,6 +20,10 @@ pipe = StableDiffusionPipeline.from_pretrained(pretrain_model, scheduler = sched
 # Apply hidiffusion with a single line of code.
 apply_hidiffusion(pipe)
 
+@spaces.GPU
+def run_hidiffusion(prompt, negative_prompt):
+    return pipe(prompt, guidance_scale=7.5, height=1024, width=1024, eta=1.0, negative_prompt=negative_prompt).images[0]
+        
 with gr.Blocks() as demo:
     prompt = gr.Textbox()
     negative_prompt = gr.Textbox()
@@ -28,8 +32,4 @@ with gr.Blocks() as demo:
 
     btn.click(fn=run_hidiffusion, inputs=[prompt, negative_prompt], outputs=[output])
     
-    @spaces.GPU
-    def run_hidiffusion(prompt, negative_prompt):
-        return pipe(prompt, guidance_scale=7.5, height=1024, width=1024, eta=1.0, negative_prompt=negative_prompt).images[0]
-
 demo.launch()