M17idd commited on
Commit
cd5ea22
·
verified ·
1 Parent(s): d382d77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -30
app.py CHANGED
@@ -181,36 +181,6 @@ st.markdown('<div class="chat-message">👋 سلام! چطور میتونم کم
181
  # st.markdown(f'<div class="chat-message">📩 شما: {user_input}</div>', unsafe_allow_html=True)
182
 
183
 
184
- """"
185
- # ----------------- لود PDF و ساخت ایندکس -----------------
186
- class TogetherEmbeddings(Embeddings):
187
- def __init__(self, model_name: str, api_key: str):
188
- self.model_name = model_name
189
- self.client = Together(api_key=api_key)
190
-
191
- def embed_documents(self, texts: List[str]) -> List[List[float]]:
192
- response = self.client.embeddings.create(model=self.model_name, input=texts)
193
- return [item.embedding for item in response.data]
194
-
195
- def embed_query(self, text: str) -> List[float]:
196
- return self.embed_documents([text])[0]
197
-
198
- @st.cache_resource
199
- def get_pdf_index():
200
- with st.spinner('📄 در حال پردازش فایل PDF...'):
201
- loader = [PyPDFLoader('test1.pdf')]
202
- embeddings = TogetherEmbeddings(
203
- model_name="togethercomputer/m2-bert-80M-8k-retrieval",
204
- api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
205
- )
206
- return VectorstoreIndexCreator(
207
- embedding=embeddings,
208
- text_splitter=RecursiveCharacterTextSplitter(chunk_size=300, chunk_overlap=0)
209
- ).from_loaders(loader)
210
-
211
- index = get_pdf_index()
212
- """
213
-
214
 
215
  # ----------------- لود csv و ساخت ایندکس -----------------
216
  class TogetherEmbeddings(Embeddings):
 
181
  # st.markdown(f'<div class="chat-message">📩 شما: {user_input}</div>', unsafe_allow_html=True)
182
 
183
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
184
 
185
  # ----------------- لود csv و ساخت ایندکس -----------------
186
  class TogetherEmbeddings(Embeddings):