orrinin commited on
Commit
a9460a2
·
verified ·
1 Parent(s): 226eb83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -31,7 +31,7 @@ gemini_embedding_model = GeminiEmbedding(api_key=api_key, model_name="models/emb
31
  Settings.llm = llm
32
  Settings.embed_model = gemini_embedding_model
33
 
34
-
35
 
36
  def extract_web(url):
37
  web_documents = SimpleWebPageReader().load_data(
@@ -90,16 +90,15 @@ def infer(message:str, history: list):
90
 
91
  if files_list:
92
  documents, option = extract_doc(files_list)
 
93
  else:
94
  if message["text"].startswith("http://") or message["text"].startswith("https://"):
95
  documents, option = extract_web(message["text"])
 
96
  elif not message["text"].startswith("http://") and not message["text"].startswith("https://") and len(history) == 0:
97
  gr.Error("Please input an url or upload file at first.")
98
 
99
 
100
- print(documents)
101
- db_path = create_col(documents)
102
-
103
  # Load from disk
104
  load_client = chromadb.PersistentClient(path=db_path)
105
 
 
31
  Settings.llm = llm
32
  Settings.embed_model = gemini_embedding_model
33
 
34
+ db_path=""
35
 
36
  def extract_web(url):
37
  web_documents = SimpleWebPageReader().load_data(
 
90
 
91
  if files_list:
92
  documents, option = extract_doc(files_list)
93
+ db_path = create_col(documents)
94
  else:
95
  if message["text"].startswith("http://") or message["text"].startswith("https://"):
96
  documents, option = extract_web(message["text"])
97
+ db_path = create_col(documents)
98
  elif not message["text"].startswith("http://") and not message["text"].startswith("https://") and len(history) == 0:
99
  gr.Error("Please input an url or upload file at first.")
100
 
101
 
 
 
 
102
  # Load from disk
103
  load_client = chromadb.PersistentClient(path=db_path)
104