Update functions.py

functions.py

@@ -151,14 +151,12 @@ def process_corpus(corpus, tok, title, embeddings, chunk_size=200, overlap=50):
     )
 
     return docsearch
-    
-@st.experimental_memo(suppress_st_warning=True)
-def embed_text(query,corpus,title,embedding_model,emb_tok,chain_type='stuff'):
-    
-    '''Embed text and generate semantic search scores'''
 
-    index_id = "earnings-embeddings"
+@st.experimental_singleton(suppress_st_warning=True)
+def gen_embeddings(embedding_model):
 
+    '''Generate embeddings for given model'''
+    
     if 'hkunlp' in embedding_model:
         
         embeddings = HuggingFaceInstructEmbeddings(model_name=f'hkunlp/{embedding_model}',
@@ -167,7 +165,18 @@ def embed_text(query,corpus,title,embedding_model,emb_tok,chain_type='stuff'):
 
     else:
         
-        embeddings = HuggingFaceEmbeddings(model_name=f'sentence-transformers/{embedding_model}')
+        embeddings = HuggingFaceEmbeddings(model_name=embedding_model)
+
+    return embeddings
+    
+@st.experimental_memo(suppress_st_warning=True)
+def embed_text(query,corpus,title,embedding_model,emb_tok,chain_type='stuff'):
+    
+    '''Embed text and generate semantic search scores'''
+
+    index_id = "earnings-embeddings"
+
+    embeddings = gen_embeddings(embedding_model)
         
     title = title[0]
     docsearch = process_corpus(corpus,embed_tok,title, embeddings)