Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import torch
 import torch.nn.functional as F
 from torch import Tensor
@@ -113,6 +114,7 @@ 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]
@@ -155,6 +157,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()