Update app.py

app.py

@@ -115,7 +115,7 @@ def build_pdf_index():
 
         documents = [LangchainDocument(page_content=t) for t in texts]
 
-        sentence_model = SentenceTransformer('HooshvareLab/bert-fa-zwnj-base')
+        sentence_model = SentenceTransformer("togethercomputer/m2-bert-80M-8k-retrieval", trust_remote_code=True)
 
         progress_bar = st.progress(0)
         total_docs = len(documents)
@@ -155,7 +155,7 @@ class SimpleRetriever(BaseRetriever):
     index: faiss.Index
 
     def _get_relevant_documents(self, query: str) -> List[Document]:
-        sentence_model = SentenceTransformer('HooshvareLab/bert-fa-zwnj-base')
+        sentence_model = SentenceTransformer("togethercomputer/m2-bert-80M-8k-retrieval", trust_remote_code=True)
         query_embedding = sentence_model.encode(query, convert_to_numpy=True)
 
         _, indices = self.index.search(np.expand_dims(query_embedding, axis=0), 5)  # پیدا کردن 5 سند مشابه