Update app.py

app.py

@@ -23,7 +23,8 @@ class ImageProcessor:
     def clear_cache(self, transformer):
         for name, attn_processor in transformer.attn_processors.items():
             attn_processor.bank_kv.clear()
-        
+            
+    @spaces.GPU()
     def process_image(self, prompt='', subject_imgs=[], spatial_imgs=[], height=768, width=768, output_path=None, seed=42):
         image = self.pipe(
             prompt,