samim2024 commited on
Commit
342cc30
·
verified ·
1 Parent(s): 8ef9874

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -51,7 +51,7 @@ def get_text(url):
51
  for paragraph in paragraphs:
52
  file.write(paragraph.get_text() + "\n")
53
 
54
- @st.cache_resource
55
  def create_langchain_index(input_text):
56
  print("--indexing---")
57
  get_text(input_text)
@@ -67,7 +67,7 @@ def create_langchain_index(input_text):
67
  persist_directory = "chroma_db"
68
  #vectordb = Chroma.from_documents(documents=docs, embedding=embeddings, persist_directory=persist_directory)
69
  #new_db = Chroma(persist_directory=persist_directory, embedding_function=embeddings)
70
- return _db
71
 
72
  # @st.cache_resource
73
  # def get_basic_page_details(input_text,summary_query,tweet_query,ln_query):
@@ -79,7 +79,7 @@ def create_langchain_index(input_text):
79
  # return summary_response,tweet_response,ln_response
80
 
81
 
82
- @st.cache_data
83
  def get_response(input_text,query,db):
84
  print(f"--querying---{query}")
85
  retrieval_chain = RetrievalQA.from_chain_type(llm, chain_type="stuff", retriever=db.as_retriever())
 
51
  for paragraph in paragraphs:
52
  file.write(paragraph.get_text() + "\n")
53
 
54
+ #@st.cache_resource
55
  def create_langchain_index(input_text):
56
  print("--indexing---")
57
  get_text(input_text)
 
67
  persist_directory = "chroma_db"
68
  #vectordb = Chroma.from_documents(documents=docs, embedding=embeddings, persist_directory=persist_directory)
69
  #new_db = Chroma(persist_directory=persist_directory, embedding_function=embeddings)
70
+ return db
71
 
72
  # @st.cache_resource
73
  # def get_basic_page_details(input_text,summary_query,tweet_query,ln_query):
 
79
  # return summary_response,tweet_response,ln_response
80
 
81
 
82
+ #@st.cache_data
83
  def get_response(input_text,query,db):
84
  print(f"--querying---{query}")
85
  retrieval_chain = RetrievalQA.from_chain_type(llm, chain_type="stuff", retriever=db.as_retriever())