Update app.py

app.py

@@ -12,6 +12,7 @@ from src.pipeline import FluxPipeline
 from src.transformer_flux import FluxTransformer2DModel
 from src.lora_helper import set_single_lora, set_multi_lora, unset_lora
 
+@spaces.GPU
 class ImageProcessor:
     def __init__(self, path):
         device = "cuda"
@@ -24,7 +25,7 @@ class ImageProcessor:
         for name, attn_processor in transformer.attn_processors.items():
             attn_processor.bank_kv.clear()
             
-    @spaces.GPU()
+    @spaces.GPU
     def process_image(self, prompt='', subject_imgs=[], spatial_imgs=[], height=768, width=768, output_path=None, seed=42):
         image = self.pipe(
             prompt,