Spaces:

kiyer
/

synthesist

Sleeping

kiyer commited on Jul 21, 2024

Commit

77b2b2e

verified ·

1 Parent(s): c0ca737

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -100,12 +100,11 @@ if 'arxiv_corpus' not in st.session_state:
     with st.spinner('loading data...'):
         try:
             arxiv_corpus = load_from_disk('data/')
-            arxiv_corpus.add_faiss_index('embed')
         except:
             st.write('downloading data')
             arxiv_corpus = load_dataset('kiyer/pathfinder_arxiv_data',split='train')
-            arxiv_corpus.add_faiss_index('embed')
             arxiv_corpus.save_to_disk('data/')
         st.session_state.arxiv_corpus = arxiv_corpus
         st.toast('loaded arxiv corpus')
 else:

     with st.spinner('loading data...'):
         try:
             arxiv_corpus = load_from_disk('data/')
         except:
             st.write('downloading data')
             arxiv_corpus = load_dataset('kiyer/pathfinder_arxiv_data',split='train')
             arxiv_corpus.save_to_disk('data/')
+        arxiv_corpus.add_faiss_index('embed')
         st.session_state.arxiv_corpus = arxiv_corpus
         st.toast('loaded arxiv corpus')
 else: