Update app.py
Browse files
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
|
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
|
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], \
|