kiyer commited on
Commit
77b2b2e
·
verified ·
1 Parent(s): c0ca737

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -100,12 +100,11 @@ if 'arxiv_corpus' not in st.session_state:
100
  with st.spinner('loading data...'):
101
  try:
102
  arxiv_corpus = load_from_disk('data/')
103
- arxiv_corpus.add_faiss_index('embed')
104
  except:
105
  st.write('downloading data')
106
  arxiv_corpus = load_dataset('kiyer/pathfinder_arxiv_data',split='train')
107
- arxiv_corpus.add_faiss_index('embed')
108
  arxiv_corpus.save_to_disk('data/')
 
109
  st.session_state.arxiv_corpus = arxiv_corpus
110
  st.toast('loaded arxiv corpus')
111
  else:
 
100
  with st.spinner('loading data...'):
101
  try:
102
  arxiv_corpus = load_from_disk('data/')
 
103
  except:
104
  st.write('downloading data')
105
  arxiv_corpus = load_dataset('kiyer/pathfinder_arxiv_data',split='train')
 
106
  arxiv_corpus.save_to_disk('data/')
107
+ arxiv_corpus.add_faiss_index('embed')
108
  st.session_state.arxiv_corpus = arxiv_corpus
109
  st.toast('loaded arxiv corpus')
110
  else: