Spaces:
Runtime error
Runtime error
Commit ·
2d178ec
1
Parent(s): c47a1d0
base model discarded
Browse files- demo_dpr.py +1 -1
demo_dpr.py
CHANGED
|
@@ -254,7 +254,7 @@ def load_models(dpr_model_path, auth_token, cross_encoder_model_path):
|
|
| 254 |
|
| 255 |
context_embeddings, contexes = load_paragraphs('st-context-embeddings.pkl')
|
| 256 |
dpr_context_embeddings, dpr_contexes = load_paragraphs('basecamp-dpr-contriever-embeddings.pkl')
|
| 257 |
-
base_dpr_context_embeddings, base_contexes = load_paragraphs('basecamp-base-dpr-contriever-embeddings.pkl')
|
| 258 |
dataframe_bsbs = load_dataframes()
|
| 259 |
dpr_trained, bi_encoder, cross_encoder, trained_cross_encoder, question_tokenizer, base_dpr_context_encoder = copy.deepcopy(load_models(st.secrets["DPR_MODEL_PATH"], st.secrets["AUTH_TOKEN"], st.secrets["CROSS_ENCODER_MODEL_PATH"]))
|
| 260 |
qa_main_widgetsv2()
|
|
|
|
| 254 |
|
| 255 |
context_embeddings, contexes = load_paragraphs('st-context-embeddings.pkl')
|
| 256 |
dpr_context_embeddings, dpr_contexes = load_paragraphs('basecamp-dpr-contriever-embeddings.pkl')
|
| 257 |
+
#base_dpr_context_embeddings, base_contexes = load_paragraphs('basecamp-base-dpr-contriever-embeddings.pkl')
|
| 258 |
dataframe_bsbs = load_dataframes()
|
| 259 |
dpr_trained, bi_encoder, cross_encoder, trained_cross_encoder, question_tokenizer, base_dpr_context_encoder = copy.deepcopy(load_models(st.secrets["DPR_MODEL_PATH"], st.secrets["AUTH_TOKEN"], st.secrets["CROSS_ENCODER_MODEL_PATH"]))
|
| 260 |
qa_main_widgetsv2()
|