EnigmaOfTheWorld commited on
Commit
73b8442
·
1 Parent(s): f0ccffb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -140,7 +140,7 @@ def search_internet(user_query:str,*,key_number:int) -> str:
140
 
141
  def search_document_uploaded(user_query:str) -> str:
142
  print('Searching uploaded document......')
143
- docsearch = FAISS.load_local(folder_path = '/tmp/embeddings',embeddings=EMBEDIDNGS)
144
  chain = load_qa_chain(OpenAI(), chain_type="stuff")
145
  docs = docsearch.similarity_search(user_query)
146
  return chain.run(input_documents=docs, question=user_query)
@@ -289,7 +289,7 @@ def build_embeddings(file_name,file_ext):
289
  docsearch = FAISS.from_texts(texts, EMBEDIDNGS)
290
  if not os.path.exists(f'/tmp/{RANDOM_USER}embeddings'):
291
  os.mkdir(f'/tmp/{RANDOM_USER}embeddings')
292
- docsearch.save_local('f/tmp/{RANDOM_USER}embeddings')
293
  print(f'Embeddings created to /tmp/{RANDOM_USER}embeddings')
294
 
295
 
 
140
 
141
  def search_document_uploaded(user_query:str) -> str:
142
  print('Searching uploaded document......')
143
+ docsearch = FAISS.load_local(folder_path = f'/tmp/{RANDOM_USER}embeddings',embeddings=EMBEDIDNGS)
144
  chain = load_qa_chain(OpenAI(), chain_type="stuff")
145
  docs = docsearch.similarity_search(user_query)
146
  return chain.run(input_documents=docs, question=user_query)
 
289
  docsearch = FAISS.from_texts(texts, EMBEDIDNGS)
290
  if not os.path.exists(f'/tmp/{RANDOM_USER}embeddings'):
291
  os.mkdir(f'/tmp/{RANDOM_USER}embeddings')
292
+ docsearch.save_local(f'/tmp/{RANDOM_USER}embeddings')
293
  print(f'Embeddings created to /tmp/{RANDOM_USER}embeddings')
294
 
295