Update app.py

app.py

@@ -114,7 +114,6 @@ def embedding_worker():
         embedding_request_queue.task_done()
         clear_cuda_cache()
 
-@spaces.GPU
 def compute_embeddings(selected_task, input_text):
     try:
         task_description = tasks[selected_task]
@@ -133,6 +132,7 @@ def compute_embeddings(selected_task, input_text):
     clear_cuda_cache()
     return embeddings_list
 
+@spaces.GPU
 def compute_similarity(selected_task, sentence1, sentence2, extra_sentence1, extra_sentence2):
     try:
         task_description = tasks[selected_task]
@@ -154,7 +154,7 @@ def compute_similarity(selected_task, sentence1, sentence2, extra_sentence1, ext
     clear_cuda_cache()
     return similarity_scores
 
-@spaces.GPU
+
 def compute_cosine_similarity(emb1, emb2):
     tensor1 = torch.tensor(emb1).to(device).half()
     tensor2 = torch.tensor(emb2).to(device).half()