Update app.py

app.py

@@ -25,6 +25,7 @@ from torchvision.transforms.functional import pil_to_tensor
 
 from metauas import MetaUAS, set_random_seed, normalize, apply_ad_scoremap, safely_load_state_dict
 from huggingface_hub import hf_hub_download
+import spaces
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
@@ -50,6 +51,7 @@ model = MetaUAS(encoder_name,
                 fusion_policy
                 )   
 
+@spaces.GPU
 def process_image(prompt_img, query_img, options):
     # Load the model based on selected options
     if 'model-512' in options: