Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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:
|