Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def sent_cross_load():
|
|
47 |
|
48 |
@st.cache
|
49 |
def load_data():
|
50 |
-
dfs = [pd.read_json(f'
|
51 |
df = pd.concat(dfs)
|
52 |
df.reset_index(inplace=True, drop=True)
|
53 |
return df
|
@@ -59,7 +59,7 @@ with st.spinner(text="Loading data..."):
|
|
59 |
|
60 |
@st.cache
|
61 |
def load_embeddings():
|
62 |
-
efs = [np.load(f'
|
63 |
corpus_embeddings = np.concatenate(efs)
|
64 |
return corpus_embeddings
|
65 |
|
|
|
47 |
|
48 |
@st.cache
|
49 |
def load_data():
|
50 |
+
dfs = [pd.read_json(f'passages_{i}.jsonl', lines=True) for i in range(0,5)]
|
51 |
df = pd.concat(dfs)
|
52 |
df.reset_index(inplace=True, drop=True)
|
53 |
return df
|
|
|
59 |
|
60 |
@st.cache
|
61 |
def load_embeddings():
|
62 |
+
efs = [np.load(f'embeddings_{i}.pt.npy') for i in range(0,5)]
|
63 |
corpus_embeddings = np.concatenate(efs)
|
64 |
return corpus_embeddings
|
65 |
|