Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
|
@@ -27,7 +27,10 @@ logger = get_logger(__name__)
|
|
| 27 |
|
| 28 |
embeddings = HuggingFaceInferenceAPIEmbeddings(
|
| 29 |
api_key=hf_api_key,
|
| 30 |
-
model_name="BAAI/bge-large-en-v1.5"
|
|
|
|
|
|
|
|
|
|
| 31 |
)
|
| 32 |
|
| 33 |
if 'chat_history' not in st.session_state:
|
|
|
|
| 27 |
|
| 28 |
embeddings = HuggingFaceInferenceAPIEmbeddings(
|
| 29 |
api_key=hf_api_key,
|
| 30 |
+
model_name="BAAI/bge-large-en-v1.5",
|
| 31 |
+
configuration: {
|
| 32 |
+
baseURL: "https://router.huggingface.co/hf-inference/pipeline/feature-extraction",
|
| 33 |
+
},
|
| 34 |
)
|
| 35 |
|
| 36 |
if 'chat_history' not in st.session_state:
|