secilozksen commited on
Commit
7321dc5
1 Parent(s): 14df537

cpu unpickler added

Browse files
Files changed (1) hide show
  1. demo_dpr.py +1 -1
demo_dpr.py CHANGED
@@ -63,7 +63,7 @@ class CPU_Unpickler(pickle.Unpickler):
63
  @st.cache(show_spinner=False, allow_output_mutation=True)
64
  def load_paragraphs(path):
65
  with open(path, "rb") as fIn:
66
- cache_data = pickle.load(fIn)
67
  corpus_sentences = cache_data['contexes']
68
  corpus_embeddings = cache_data['embeddings']
69
 
 
63
  @st.cache(show_spinner=False, allow_output_mutation=True)
64
  def load_paragraphs(path):
65
  with open(path, "rb") as fIn:
66
+ cache_data = CPU_Unpickler(fIn).load(fIn)
67
  corpus_sentences = cache_data['contexes']
68
  corpus_embeddings = cache_data['embeddings']
69