M17idd commited on
Commit
162eb65
·
verified ·
1 Parent(s): e495c4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -111,8 +111,8 @@ tokenizer.pad_token = tokenizer.eos_token # یا می‌توانید این خ
111
  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="HooshvareLab/gpt2-fa", tokenizer=tokenizer)
116
  index = VectorstoreIndexCreator(embedding=embeddings, text_splitter=RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=128)).from_loaders([pdf_loader])
117
  return index
118
 
 
111
  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="PartAI/Dorna-Llama3-8B-Instruct")
116
  index = VectorstoreIndexCreator(embedding=embeddings, text_splitter=RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=128)).from_loaders([pdf_loader])
117
  return index
118