Spaces:
Build error
Build error
fixed bge_model_name_notfound
Browse fileshf threw an error when loading HFBGEEmbeddings
- 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 |
-
|
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
|