Update app.py
Browse files
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=
|
| 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':
|
| 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())
|