use test_sub.jsonl

app.py

@@ -36,7 +36,7 @@ def jsonl_to_df(file_path):
 
 @st.cache
 def load_test_df():
-    file_path = os.path.join('data', 'nela_gt_2018_site_split', 'test.jsonl')
+    file_path = os.path.join('test_sub.jsonl')
     test_df = jsonl_to_df(file_path)
     test_df = pd.get_dummies(test_df, columns = ['label'])
     return test_df
@@ -90,8 +90,6 @@ def correct_preds(preds, labels):
 
 
 if __name__ == '__main__':
-    if not os.path.exists('data/nela_gt_2018_site_split/test.jsonl'):
-        download_dataset()
     df = load_test_df()
     tokenizer, model = init_model()