Spaces:
Runtime error
Runtime error
secilozksen
commited on
Commit
•
7321dc5
1
Parent(s):
14df537
cpu unpickler added
Browse files- 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 =
|
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 |
|