Update app.py

app.py

@@ -33,6 +33,10 @@ class Config:
     
 xflux_pipeline = XFluxPipeline(Config.name, Config.device, Config.offload)
 
+@spaces.GPU
+def generate(**kwargs): 
+    return xflux_pipeline.gradio_generate(**kwargs)
+    
 def parse_args() -> Config:
     parser = argparse.ArgumentParser(description="Flux")
     parser.add_argument("--name", type=str, default="flux-dev", help="Model name")
@@ -201,7 +205,7 @@ def create_demo(
                     lora_weight, local_path, lora_local_path, ip_local_path
                     ]
             generate_btn.click(
-                fn=xflux_pipeline.gradio_generate,
+                fn=generate,
                 inputs=inputs,
                 outputs=[output_image, download_btn],
             )