Update functions.py

functions.py

@@ -167,7 +167,7 @@ def embed_text(query,corpus,title,embedding_model,_emb_tok,_chain_type):
 
     print(docs)
 
-    if chain_type == 'Normal':
+    if _chain_type == 'Normal':
 
         docs = [d[0] for d in docs]
 
@@ -183,7 +183,7 @@ def embed_text(query,corpus,title,embedding_model,_emb_tok,_chain_type):
         answer = chain({"input_documents": docs, "question": query}, return_only_outputs=False)
 
 
-    elif chain_type == 'Refined':
+    elif _chain_type == 'Refined':
 
         docs = [d[0] for d in docs]