Spaces:

obss
/

question-answering-demo

Runtime error

secilozksen commited on Feb 23, 2023

Commit

c47a1d0

•

1 Parent(s): 7321dc5

Load bug fix

Files changed (1) hide show

demo_dpr.py CHANGED Viewed

@@ -63,7 +63,7 @@ class CPU_Unpickler(pickle.Unpickler):
 @st.cache(show_spinner=False, allow_output_mutation=True)
 def load_paragraphs(path):
     with open(path, "rb") as fIn:
-        cache_data = CPU_Unpickler(fIn).load(fIn)
         corpus_sentences = cache_data['contexes']
         corpus_embeddings = cache_data['embeddings']

 @st.cache(show_spinner=False, allow_output_mutation=True)
 def load_paragraphs(path):
     with open(path, "rb") as fIn:
+        cache_data = CPU_Unpickler(fIn).load()
         corpus_sentences = cache_data['contexes']
         corpus_embeddings = cache_data['embeddings']