Nelly-43 commited on
Commit
02ab93e
·
verified ·
1 Parent(s): fca1e2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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 "BAAI/bge-base-en-v1.5"
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
- login_hug(os.getenv('HF_TOKEN'))
859
- embedding_name=gr.Dropdown(choices=["BAAI/bge-base-en-v1.5","dunzhang/stella_en_1.5B_v5","dunzhang/stella_en_400M_v5",
 
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"],