M17idd commited on
Commit
200914c
·
1 Parent(s): e5b04f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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="sentence-transformers/all-MiniLM-L6-v2"
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="HuggingFaceH4/zephyr-7b-alpha", # یا مدل دلخواهت مثل llama یا mistral
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(