Adding else block to skip

app.py

@@ -11,7 +11,7 @@ contribution_json = "contributors.json"
 contribution_dict = json.load(open(contribution_json,"r"))
 
 
-splits = ['EuroParliamentProceedings', 'TED2020', 'PileOfLaw', 'StackExchange_ver2', 'GithubIssues', 'Opensubtitles', 'USPTO', 'S2ORC', 'DevDocs', 'CodePileReddit2022', 'DMMath', 'Gutenberg', 'USENET', 'GithubDiff', 'Enwiki', 'GNOME', 'ASFPublicMail', 'PileV2Reddit2020', 'CodePilePosts', 'Discourse', 'Tanzil', 'arXiv', 'UbuntuIRC', 'PubMed', 'CodePileReddit2020', 'CodePileReddit2021', 'GlobalVoices', 'FreeLaw_Options', 'PileV2Posts','Bible']
+splits = os.listdir(CACHE_DIR)#['EuroParliamentProceedings', 'TED2020', 'PileOfLaw', 'StackExchange_ver2', 'GithubIssues', 'Opensubtitles', 'USPTO', 'S2ORC', 'DevDocs', 'CodePileReddit2022', 'DMMath', 'Gutenberg', 'USENET', 'GithubDiff', 'Enwiki', 'GNOME', 'ASFPublicMail', 'PileV2Reddit2020', 'CodePilePosts', 'Discourse', 'Tanzil', 'arXiv', 'UbuntuIRC', 'PubMed', 'CodePileReddit2020', 'CodePileReddit2021', 'GlobalVoices', 'FreeLaw_Options', 'PileV2Posts','Bible']
 
 cached_ds = os.listdir(CACHE_DIR)
 tokenizer = AutoTokenizer.from_pretrained('EleutherAI/gpt-neox-20b')
@@ -26,7 +26,10 @@ def load_page(split):
     print("Sucessfully loaded "+split)
     st.title("Dataset Explorer")
     st.write(f"# {split}")
-    st.caption(f"Contributors: {','.join(contribution_dict[split])}")
+    if split in contribution_dict:
+        st.caption(f"Contributors: {','.join(contribution_dict[split])}")
+    else:
+        st.caption(f"Needs to be updated....")
     with st.form("dataset_form"):
         index = st.slider('Select a row', 0, len(ds)-1, 0)
         if st.form_submit_button("Load"):
@@ -46,12 +49,4 @@ def load_page(split):
 
 
 demo_name = st.sidebar.selectbox("Choose a demo", splits)
-load_page(demo_name)
-    # st.write(f"Loaded {ds} with {len(dataset['train'])} rows")
-
-# st.sidebar.title('Pile v2 Explorer')
-# split = st.sidebar.selectbox('Select a split', splits)
-# st.sidebar.write('You selected:', split)
-# dataset = datasets.load_dataset('CarperAI/pile-v2-small-filtered', data_dir="data/"+split+"/data.json")
-# index = st.sidebar.slider('Select a row', 0, len(dataset['train'])-1, 0)
-# st.write(dataset['train'][index]['text'])
+load_page(demo_name)