focusit commited on
Commit
ef5b95e
·
1 Parent(s): 931948b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,14 +13,14 @@ hub_token = os.environ["hub_key"]
13
  # Load text
14
  loader = TextLoader("testing.txt")
15
  documents = loader.load()
16
- splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
17
  docs = splitter.split_documents(documents)
18
 
19
  embeddings = HuggingFaceEmbeddings()
20
  doc_search = Chroma.from_documents(docs, embeddings)
21
 
22
  repo_id = "tiiuae/falcon-7b"
23
- llm = HuggingFaceHub(repo_id=repo_id, huggingfacehub_api_token=hub_token, model_kwargs={'temperature': 0.2,'min_length': 1000})
24
 
25
  from langchain.schema import retriever
26
  retireval_chain = RetrievalQA.from_chain_type(llm, chain_type="stuff", retriever=doc_search.as_retriever())
 
13
  # Load text
14
  loader = TextLoader("testing.txt")
15
  documents = loader.load()
16
+ splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=20)
17
  docs = splitter.split_documents(documents)
18
 
19
  embeddings = HuggingFaceEmbeddings()
20
  doc_search = Chroma.from_documents(docs, embeddings)
21
 
22
  repo_id = "tiiuae/falcon-7b"
23
+ llm = HuggingFaceHub(repo_id=repo_id, huggingfacehub_api_token=hub_token, model_kwargs={'temperature': 0.2,'min_length': 6000})
24
 
25
  from langchain.schema import retriever
26
  retireval_chain = RetrievalQA.from_chain_type(llm, chain_type="stuff", retriever=doc_search.as_retriever())