M17idd commited on
Commit
59fbf80
·
verified ·
1 Parent(s): e595e00

Update app.py

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