Update functions.py

functions.py

@@ -194,7 +194,6 @@ def embed_text(query,corpus,title,embedding_model,_emb_tok,chain_type='stuff'):
 
         answer = chain({"input_documents": docs, "question": query}, return_only_outputs=False)
 
-        return answer
 
     elif chain_type == 'Refined':
 
@@ -207,7 +206,7 @@ def embed_text(query,corpus,title,embedding_model,_emb_tok,chain_type='stuff'):
                      question_prompt=initial_qa_prompt, refine_prompt=refine_prompt)
         answer = chain({"input_documents": docs, "question": query}, return_only_outputs=False)
 
-        return answer
+    return answer
     
 @st.experimental_singleton(suppress_st_warning=True)
 def get_spacy():