NavyDevilDoc commited on
Commit
1d8bd27
·
verified ·
1 Parent(s): d23aae5

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +2 -2
src/app.py CHANGED
@@ -310,8 +310,8 @@ with st.sidebar:
310
  embed_options = {
311
  "Standard (All-MiniLM, 384d)": "sentence-transformers/all-MiniLM-L6-v2",
312
  "High-Perf (MPNet, 768d)": "sentence-transformers/all-mpnet-base-v2",
313
- "OpenAI Small (1536d)": "text-embedding-3-small"
314
- "Custom Navy (BGE, 768d)": "NavyDevilDoc/navy-custom-models/bge-finetuned" # NEW
315
  }
316
  embed_choice_label = st.selectbox("Select Embedding Model", list(embed_options.keys()))
317
  st.session_state.active_embed_model = embed_options[embed_choice_label]
 
310
  embed_options = {
311
  "Standard (All-MiniLM, 384d)": "sentence-transformers/all-MiniLM-L6-v2",
312
  "High-Perf (MPNet, 768d)": "sentence-transformers/all-mpnet-base-v2",
313
+ "OpenAI Small (1536d)": "text-embedding-3-small",
314
+ "Custom Navy (BGE, 768d)": "NavyDevilDoc/navy-custom-models/bge-finetuned"
315
  }
316
  embed_choice_label = st.selectbox("Select Embedding Model", list(embed_options.keys()))
317
  st.session_state.active_embed_model = embed_options[embed_choice_label]