secilozksen commited on
Commit
2d178ec
1 Parent(s): c47a1d0

base model discarded

Browse files
Files changed (1) hide show
  1. 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()