satyam001 commited on
Commit
d5b5d7d
·
1 Parent(s): ec77767

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- huggingfacehub_api_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)
 
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)