nickmuchi commited on
Commit
14ec610
1 Parent(s): de88316

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +2 -4
functions.py CHANGED
@@ -155,14 +155,12 @@ def gen_embeddings(embedding_model):
155
  return embeddings
156
 
157
  @st.experimental_memo(suppress_st_warning=True)
158
- def embed_text(query,corpus,title,embedding_model,_emb_tok,chain_type):
159
 
160
  '''Embed text and generate semantic search scores'''
161
 
162
  title = title.split()[0].lower()
163
-
164
- docsearch = process_corpus(corpus,_emb_tok,title, embedding_model)
165
-
166
  docs = docsearch.similarity_search_with_score(query, k=3)
167
 
168
  if chain_type == 'Normal':
 
155
  return embeddings
156
 
157
  @st.experimental_memo(suppress_st_warning=True)
158
+ def embed_text(query,corpus,title,embedding_model,_emb_tok,docsearch,chain_type):
159
 
160
  '''Embed text and generate semantic search scores'''
161
 
162
  title = title.split()[0].lower()
163
+
 
 
164
  docs = docsearch.similarity_search_with_score(query, k=3)
165
 
166
  if chain_type == 'Normal':