Update app.py

app.py

@@ -122,7 +122,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("aubmindlab/bert-base-arabert")
 
         progress_bar = st.progress(0)
         total_docs = len(documents)
@@ -162,7 +162,7 @@ class SimpleRetriever(BaseRetriever):
 
     def _get_relevant_documents(self, query: str) -> List[Document]:
         # فقط از sentence_model استفاده می‌کنیم
-        sentence_model = SentenceTransformer('HooshvareLab/bert-fa-zwnj-base')
+        sentence_model = SentenceTransformer("aubmindlab/bert-base-arabert")
         query_embedding = sentence_model.encode(query, convert_to_numpy=True)
 
         similarities = []