Jack Sambath commited on
Commit
786ffe1
·
1 Parent(s): 21a4b65
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def main():
18
  embeddings = HuggingFaceEmbeddings()
19
  vectordb = Chroma.from_documents(documents=texts,embedding=embeddings)
20
  # llm = HuggingFaceHub(repo_id="google/flan-t5-small", model_kwargs={"temperature":0,"max_length":200}, huggingfacehub_api_token="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") # type: ignore
21
- qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff",retriever=vectordb.as_retriever(search_type="mmr", search_kwargs={'fetch_k': 30}), return_source_documents=True)
22
  query = st.text_input("Ask a question: ")
23
  qa(inputs=query,return_only_outputs=True)
24
  result = qa({"query": query})
 
18
  embeddings = HuggingFaceEmbeddings()
19
  vectordb = Chroma.from_documents(documents=texts,embedding=embeddings)
20
  # llm = HuggingFaceHub(repo_id="google/flan-t5-small", model_kwargs={"temperature":0,"max_length":200}, huggingfacehub_api_token="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") # type: ignore
21
+ qa = RetrievalQA.from_chain_type(llm=llm(), chain_type="stuff",retriever=vectordb.as_retriever(search_type="mmr", search_kwargs={'fetch_k': 30}), return_source_documents=True)
22
  query = st.text_input("Ask a question: ")
23
  qa(inputs=query,return_only_outputs=True)
24
  result = qa({"query": query})