Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ from langchain.chains import RetrievalQA
|
|
9 |
from langchain.chat_models import ChatOpenAI
|
10 |
from typing import List
|
11 |
from together import Together
|
|
|
12 |
|
13 |
st.set_page_config(page_title="چت بات ارتش", page_icon="🪖", layout="wide")
|
14 |
|
@@ -88,7 +89,7 @@ def get_pdf_index():
|
|
88 |
texts = text_splitter.split_documents(documents)
|
89 |
|
90 |
embeddings = HuggingFaceEmbeddings(
|
91 |
-
model_name="
|
92 |
)
|
93 |
|
94 |
index = FAISS.from_documents(texts, embeddings)
|
@@ -97,9 +98,8 @@ def get_pdf_index():
|
|
97 |
index = get_pdf_index()
|
98 |
|
99 |
llm = HuggingFaceHub(
|
100 |
-
repo_id="
|
101 |
-
model_kwargs={"temperature": 0.7, "max_new_tokens": 512}
|
102 |
-
huggingfacehub_api_token="توکن شما"
|
103 |
)
|
104 |
|
105 |
chain = RetrievalQA.from_chain_type(
|
|
|
9 |
from langchain.chat_models import ChatOpenAI
|
10 |
from typing import List
|
11 |
from together import Together
|
12 |
+
from langchain.embeddings import HuggingFaceEmbeddings
|
13 |
|
14 |
st.set_page_config(page_title="چت بات ارتش", page_icon="🪖", layout="wide")
|
15 |
|
|
|
89 |
texts = text_splitter.split_documents(documents)
|
90 |
|
91 |
embeddings = HuggingFaceEmbeddings(
|
92 |
+
model_name="deepseek-ai/DeepSeek-R1"
|
93 |
)
|
94 |
|
95 |
index = FAISS.from_documents(texts, embeddings)
|
|
|
98 |
index = get_pdf_index()
|
99 |
|
100 |
llm = HuggingFaceHub(
|
101 |
+
repo_id="deepseek-ai/DeepSeek-R1",
|
102 |
+
model_kwargs={"temperature": 0.7, "max_new_tokens": 512}
|
|
|
103 |
)
|
104 |
|
105 |
chain = RetrievalQA.from_chain_type(
|