isayahc commited on
Commit
e06901a
Β·
1 Parent(s): 7c39d08

Update app.py

Browse files

fo user to download from gdown

Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -17,9 +17,15 @@ from langchain.chains import RetrievalQA
17
 
18
  from langchain.prompts import ChatPromptTemplate
19
 
 
 
20
  #web_links = ["https://www.databricks.com/","https://help.databricks.com","https://docs.databricks.com","https://kb.databricks.com/","http://docs.databricks.com/getting-started/index.html","http://docs.databricks.com/introduction/index.html","http://docs.databricks.com/getting-started/tutorials/index.html","http://docs.databricks.com/machine-learning/index.html","http://docs.databricks.com/sql/index.html"]
21
  #loader = WebBaseLoader(web_links)
22
  #documents = loader.load()
 
 
 
 
23
 
24
  db = Chroma(persist_directory="./chroma_db", embedding_function=embeddings)
25
  db.get()
 
17
 
18
  from langchain.prompts import ChatPromptTemplate
19
 
20
+ from utils import download_from_google_drive
21
+
22
  #web_links = ["https://www.databricks.com/","https://help.databricks.com","https://docs.databricks.com","https://kb.databricks.com/","http://docs.databricks.com/getting-started/index.html","http://docs.databricks.com/introduction/index.html","http://docs.databricks.com/getting-started/tutorials/index.html","http://docs.databricks.com/machine-learning/index.html","http://docs.databricks.com/sql/index.html"]
23
  #loader = WebBaseLoader(web_links)
24
  #documents = loader.load()
25
+
26
+ gdown_file_id = os.getenv(gdown_file_id)
27
+
28
+ download_from_google_drive(gdown_file_id)
29
 
30
  db = Chroma(persist_directory="./chroma_db", embedding_function=embeddings)
31
  db.get()