Spaces:

M17idd
/

army

Running

M17idd commited on 8 days ago

Commit

0bcc41f

verified ·

1 Parent(s): 3b8436b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -97,7 +97,7 @@ with col2:
 st.markdown("""
     <div class="header-text">
         <h1>چت‌ بات توانا</h1>
-        <div class="subtitle">دستیار هوشمند برای تصمیم‌گیری در میدان نبرد</div>
     </div>
 """, unsafe_allow_html=True)
@@ -106,7 +106,7 @@ st.markdown("""
 def get_pdf_index():
     with st.spinner('📄 در حال پردازش فایل PDF...'):
         pdf_reader = [PyPDFLoader('test1.pdf')]
-        embeddings = HuggingFaceInstructEmbeddings(model_name="universitytehran/PersianMind-v1.0")
         index  = VectorstoreIndexCreator( embedding=embeddings, text_splitter=RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=128)).from_loaders(pdf_reader)
         return index

 st.markdown("""
     <div class="header-text">
         <h1>چت‌ بات توانا</h1>
+        <div class="subtitle">دستیار هوشمند</div>
     </div>
 """, unsafe_allow_html=True)
 def get_pdf_index():
     with st.spinner('📄 در حال پردازش فایل PDF...'):
         pdf_reader = [PyPDFLoader('test1.pdf')]
+        embeddings = HuggingFaceInstructEmbeddings(model_name="HooshvareLab/gpt2-fa")
         index  = VectorstoreIndexCreator( embedding=embeddings, text_splitter=RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=128)).from_loaders(pdf_reader)
         return index