Update functions.py

functions.py

@@ -21,6 +21,7 @@ import pickle, math
 import wikipedia
 from pyvis.network import Network
 import torch
+from InstructorEmbedding import INSTRUCTOR
 
 nltk.download('punkt')
 
@@ -54,7 +55,10 @@ def load_asr_model(asr_model_name):
         
 @st.experimental_singleton(suppress_st_warning=True)    
 def load_sbert(model_name):
-    sbert = SentenceTransformer(model_name)
+    if 'hkunlp' in model_name:
+        sbert = INSTRUCTOR(model_name)
+    else:
+        sbert = SentenceTransformer(model_name)
     
     return sbert