Update app.py
Browse files
app.py
CHANGED
|
@@ -844,7 +844,7 @@ with gr.Blocks(title="LLM QA Chatbot Builder") as demo:
|
|
| 844 |
saved_params = load_params_from_file()
|
| 845 |
# If saved parameters exist, use them; otherwise, set default values
|
| 846 |
default_model_name = saved_params['model_name'] if saved_params else "Llama"
|
| 847 |
-
default_embedding_name = saved_params['embedding_name'] if saved_params else "
|
| 848 |
default_splitter_type = saved_params['splitter_type_dropdown'] if saved_params else "character"
|
| 849 |
default_chunk_size = saved_params['chunk_size_slider'] if saved_params else 500
|
| 850 |
default_chunk_overlap = saved_params['chunk_overlap_slider'] if saved_params else 30
|
|
@@ -855,8 +855,9 @@ with gr.Blocks(title="LLM QA Chatbot Builder") as demo:
|
|
| 855 |
def login_hug(token):
|
| 856 |
from huggingface_hub import login
|
| 857 |
login(token=token)
|
| 858 |
-
|
| 859 |
-
|
|
|
|
| 860 |
"nvidia/NV-Embed-v2","Alibaba-NLP/gte-Qwen2-1.5B-instruct"],value=default_embedding_name,
|
| 861 |
label="Select the Embedding Model")
|
| 862 |
splitter_type_dropdown = gr.Dropdown(choices=["character", "recursive", "token"],
|
|
|
|
| 844 |
saved_params = load_params_from_file()
|
| 845 |
# If saved parameters exist, use them; otherwise, set default values
|
| 846 |
default_model_name = saved_params['model_name'] if saved_params else "Llama"
|
| 847 |
+
default_embedding_name = saved_params['embedding_name'] if saved_params else "sentence-transformers/all-mpnet-base-v2"
|
| 848 |
default_splitter_type = saved_params['splitter_type_dropdown'] if saved_params else "character"
|
| 849 |
default_chunk_size = saved_params['chunk_size_slider'] if saved_params else 500
|
| 850 |
default_chunk_overlap = saved_params['chunk_overlap_slider'] if saved_params else 30
|
|
|
|
| 855 |
def login_hug(token):
|
| 856 |
from huggingface_hub import login
|
| 857 |
login(token=token)
|
| 858 |
+
token_tb=gr.Textbox(label="Enter your Huggingface token to access Huggingface models")
|
| 859 |
+
token_tb.change(login_hug,token_tb,None)
|
| 860 |
+
embedding_name=gr.Dropdown(choices=["sentence-transformers/all-mpnet-base-v2", "BAAI/bge-base-en-v1.5","dunzhang/stella_en_1.5B_v5","dunzhang/stella_en_400M_v5",
|
| 861 |
"nvidia/NV-Embed-v2","Alibaba-NLP/gte-Qwen2-1.5B-instruct"],value=default_embedding_name,
|
| 862 |
label="Select the Embedding Model")
|
| 863 |
splitter_type_dropdown = gr.Dropdown(choices=["character", "recursive", "token"],
|