Update app.py

app.py

@@ -10,6 +10,7 @@ import open_clip
 from huggingface_hub import hf_hub_download
 from IP_Composer.IP_Adapter.ip_adapter import IPAdapterXL
 from IP_Composer.perform_swap import compute_dataset_embeds_svd, get_modified_images_embeds_composition
+import spaces
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -36,6 +37,7 @@ CONCEPTS_MAP = {'age':'age','animal fur':'animal_fur', 'deterioration': 'deterio
 
 concept_options = list(CONCEPTS_MAP.keys())
 
+@spaces.GPU
 def get_image_embeds(pil_image, model=clip_model, preproc=preprocess, dev=device):
     """Get CLIP image embeddings for a given PIL image"""
     image = preproc(pil_image)[np.newaxis, :, :, :]
@@ -43,6 +45,7 @@ def get_image_embeds(pil_image, model=clip_model, preproc=preprocess, dev=device
         embeds = model.encode_image(image.to(dev))
     return embeds.cpu().detach().numpy()
 
+@spaces.GPU
 def process_images(
     base_image, 
     concept_image1, concept_name1,