plebias commited on
Commit
6ff6859
·
verified ·
1 Parent(s): 5e492a8

fixed bge_model_name_notfound

Browse files

hf threw an error when loading HFBGEEmbeddings

Files changed (1) hide show
  1. app_final.py +1 -1
app_final.py CHANGED
@@ -92,7 +92,7 @@ model_kwargs = {"device": "cpu"}
92
  encode_kwargs = {"normalize_embeddings": True}
93
  if "embeddings" not in st.session_state:
94
  st.session_state.embeddings = HuggingFaceBgeEmbeddings(
95
- # model_name=model_name,
96
  model_kwargs = model_kwargs,
97
  encode_kwargs = encode_kwargs)
98
  embeddings = st.session_state.embeddings
 
92
  encode_kwargs = {"normalize_embeddings": True}
93
  if "embeddings" not in st.session_state:
94
  st.session_state.embeddings = HuggingFaceBgeEmbeddings(
95
+ model_name=model_name,
96
  model_kwargs = model_kwargs,
97
  encode_kwargs = encode_kwargs)
98
  embeddings = st.session_state.embeddings