ubermenchh commited on
Commit
4eb428c
1 Parent(s): 56f68b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def paper_changes(paper_id):
18
  documents = loader.load()
19
  text_splitter = CharacterTextSplitter(chunk_size=1024, chunk_overlap=0)
20
  texts = text_splitter.split_documents(documents)
21
- embeddings = HuggingFaceEmbeddings("sentence-transformers/all-mpnet-base-v2")
22
  db = Chroma.from_documents(texts, embeddings, persist_directory="chroma_db")
23
  retriever = db.as_retriever()
24
  llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={'temperature': 0.5, 'max_new_tokens': 2096})
 
18
  documents = loader.load()
19
  text_splitter = CharacterTextSplitter(chunk_size=1024, chunk_overlap=0)
20
  texts = text_splitter.split_documents(documents)
21
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
22
  db = Chroma.from_documents(texts, embeddings, persist_directory="chroma_db")
23
  retriever = db.as_retriever()
24
  llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={'temperature': 0.5, 'max_new_tokens': 2096})