Upload app.py

app.py

@@ -415,13 +415,17 @@ def create_demo(
                 output_image = gr.Image(label="Generated Image")
                 download_btn = gr.File(label="Download full-resolution")
 
+        @spaces.GPU()
+        def gradio_generate(*args):
+            return xflux_pipeline.gradio_generate(*args)
+        
         inputs = [prompt, image_prompt, controlnet_image, width, height, guidance,
                   num_steps, seed, true_gs, ip_scale, neg_ip_scale, neg_prompt,
                   neg_image_prompt, timestep_to_start_cfg, control_type, control_weight,
                   lora_weight, local_path, lora_local_path, ip_local_path
                   ]
         generate_btn.click(
-            fn=xflux_pipeline.gradio_generate,
+            fn=gradio_generate,
             inputs=inputs,
             outputs=[output_image, download_btn],
         )