Spaces:
Sleeping
Sleeping
Commit
·
4eb428c
1
Parent(s):
56f68b0
Update app.py
Browse files
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})
|