Update app.py

app.py

@@ -8,7 +8,9 @@ from typing import List
 import zipfile
 import os
 import io
+import spaces
 
+@spaces.GPU
 def calculate_similarity(code1, code2, Ws, Wl, Wj, model_name):
     model = SentenceTransformer(model_name)
     embedding1 = model.encode(code1)
@@ -49,7 +51,8 @@ def perform_paraphrase_mining(model, codes_list, weight_semantic, weight_levensh
         weight_levenshtein=weight_levenshtein,
         weight_jaro_winkler=weight_jaro_winkler
     )
-    
+
+@spaces.GPU    
 def paraphrase_mining_with_combined_score(
     model,
     sentences: List[str],