Update app.py
Browse files
app.py
CHANGED
|
@@ -112,7 +112,16 @@ def get_pdf_index():
|
|
| 112 |
with st.spinner('📄 در حال پردازش فایل PDF...'):
|
| 113 |
pdf_loader = PyPDFLoader('test1.pdf')
|
| 114 |
# embeddings = SentenceTransformer("Thomslionel/embedings")
|
| 115 |
-
embeddings = HuggingFaceInstructEmbeddings(model_name="aidal/Persian-Mistral-7B")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 116 |
index = VectorstoreIndexCreator(embedding=embeddings, text_splitter=RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=128)).from_loaders([pdf_loader])
|
| 117 |
return index
|
| 118 |
|
|
|
|
| 112 |
with st.spinner('📄 در حال پردازش فایل PDF...'):
|
| 113 |
pdf_loader = PyPDFLoader('test1.pdf')
|
| 114 |
# embeddings = SentenceTransformer("Thomslionel/embedings")
|
| 115 |
+
# embeddings = HuggingFaceInstructEmbeddings(model_name="aidal/Persian-Mistral-7B")
|
| 116 |
+
|
| 117 |
+
embeddings = TogetherEmbeddings(
|
| 118 |
+
model_name="togethercomputer/m2-bert-80M-8k-retrieval",
|
| 119 |
+
api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
|
| 120 |
+
|
| 121 |
+
|
| 122 |
+
|
| 123 |
+
|
| 124 |
+
)
|
| 125 |
index = VectorstoreIndexCreator(embedding=embeddings, text_splitter=RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=128)).from_loaders([pdf_loader])
|
| 126 |
return index
|
| 127 |
|