vishwask commited on
Commit
0bdf2cb
·
verified ·
1 Parent(s): 5465355

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -133,7 +133,7 @@ def initialize_database(list_file_obj, chunk_size, chunk_overlap, progress=gr.Pr
133
  # global vector_db
134
  vector_db = create_db(doc_splits, collection_name)
135
  progress(0.9, desc="Done!")
136
- return vector_db, collection_name, "Complete!"
137
 
138
 
139
  def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, progress=gr.Progress()):
@@ -220,7 +220,7 @@ def demo():
220
  #upload_btn.upload(upload_file, inputs=[upload_btn], outputs=[document])
221
  db_btn.click(initialize_database, \
222
  inputs=[document, slider_chunk_size, slider_chunk_overlap], \
223
- outputs=[vector_db, collection_name, db_progress])
224
  qachain_btn.click(initialize_LLM, \
225
  inputs=[llm_model, slider_temperature, slider_maxtokens, slider_topk, vector_db], \
226
  outputs=[qa_chain, llm_progress]).then(lambda:[None,"",0,"",0,"",0], \
 
133
  # global vector_db
134
  vector_db = create_db(doc_splits, collection_name)
135
  progress(0.9, desc="Done!")
136
+ return vector_db, collection_name
137
 
138
 
139
  def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, progress=gr.Progress()):
 
220
  #upload_btn.upload(upload_file, inputs=[upload_btn], outputs=[document])
221
  db_btn.click(initialize_database, \
222
  inputs=[document, slider_chunk_size, slider_chunk_overlap], \
223
+ outputs=[vector_db, collection_name])
224
  qachain_btn.click(initialize_LLM, \
225
  inputs=[llm_model, slider_temperature, slider_maxtokens, slider_topk, vector_db], \
226
  outputs=[qa_chain, llm_progress]).then(lambda:[None,"",0,"",0,"",0], \