Update app.py
Browse files
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 |
-
|
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 |
|