Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -68,7 +68,7 @@ def main():
|
|
68 |
# embeddings = OpenAIEmbeddings(
|
69 |
# openai_api_key='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')
|
70 |
# embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
|
71 |
-
embeddings = HuggingFaceHubEmbeddings()
|
72 |
VectorStore = FAISS.from_texts(chunks, embedding=embeddings)
|
73 |
# VectorStore=Chroma.from_documents(chunks, embeddings)
|
74 |
with open(f"{store_name}.pkl", "wb") as f:
|
@@ -84,7 +84,7 @@ def main():
|
|
84 |
if query:
|
85 |
docs = VectorStore.similarity_search(query=query, k=3)
|
86 |
llm = HuggingFaceHub(repo_id='OpenAssistant/oasst-sft-1-pythia-12b',
|
87 |
-
|
88 |
# llm = OpenAI()
|
89 |
chain = load_qa_chain(llm=llm, chain_type="stuff")
|
90 |
response = chain.run(input_documents=docs, question=query)
|
|
|
68 |
# embeddings = OpenAIEmbeddings(
|
69 |
# openai_api_key='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')
|
70 |
# embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
|
71 |
+
embeddings = HuggingFaceHubEmbeddings(huggingfacehub_api_token='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')
|
72 |
VectorStore = FAISS.from_texts(chunks, embedding=embeddings)
|
73 |
# VectorStore=Chroma.from_documents(chunks, embeddings)
|
74 |
with open(f"{store_name}.pkl", "wb") as f:
|
|
|
84 |
if query:
|
85 |
docs = VectorStore.similarity_search(query=query, k=3)
|
86 |
llm = HuggingFaceHub(repo_id='OpenAssistant/oasst-sft-1-pythia-12b',
|
87 |
+
token='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', model_kwargs={"temperature": 0.1, "max_new_tokens": 250})
|
88 |
# llm = OpenAI()
|
89 |
chain = load_qa_chain(llm=llm, chain_type="stuff")
|
90 |
response = chain.run(input_documents=docs, question=query)
|