NealCaren commited on
Commit
642f7f2
1 Parent(s): a17e0df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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'data/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,7 +59,7 @@ with st.spinner(text="Loading data..."):
59
 
60
  @st.cache
61
  def load_embeddings():
62
- efs = [np.load(f'data/embeddings_{i}.pt.npy') for i in range(0,5)]
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