Spaces:
Runtime error
Runtime error
secilozksen
commited on
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()
|