Update utils.py

utils.py

@@ -56,6 +56,12 @@ def cosineSim(text1, text2):
     cosine = get_cosine(vector1, vector2)
     return cosine
 
+def cos_sim_torch(embedding_1, embedding_2):
+    return util.pytorch_cos_sim(embedding_1, embedding_2).item()
+
+def embed_text(text):
+    return model.encode(text, convert_to_tensor=True)
+
 def sentence_similarity(text1, text2):
     embedding_1= model.encode(text1, convert_to_tensor=True)
     embedding_2 = model.encode(text2, convert_to_tensor=True)
@@ -215,11 +221,11 @@ def matchingScoreWithTimeout(sentence, content):
     def timeout_handler():
         raise TimeoutError("Function timed out")
 
-    timer = threading.Timer(2, timeout_handler)  # Set a timer for 2 seconds
+    timer = threading.Timer(10, timeout_handler)  # Set a timer for 2 seconds
     timer.start()
     try:
-        # score = sentence_similarity(sentence, content)
-        score = matchingScore(sentence, content)
+        score = sentence_similarity(sentence, content)
+        # score = matchingScore(sentence, content)
         timer.cancel()  # Cancel the timer if calculation completes before timeout
         return score
     except TimeoutError: