Update app.py

app.py

@@ -5,19 +5,17 @@ import pandas as pd
 from findfile import find_files
 
 from pyabsa import ATEPCCheckpointManager
-from pyabsa.functional.dataset.dataset_manager import download_datasets_from_github, ABSADatasetList
+from pyabsa.functional.dataset.dataset_manager import download_datasets_from_github, ABSADatasetList, detect_infer_dataset
 
 download_datasets_from_github(os.getcwd())
 
+dataset_items = {dataset.name: dataset for dataset in ABSADatasetList()}
 
 def get_example(dataset):
-    filter_key_words = ['.py', '.md', 'readme', 'log', 'result', 'zip', '.state_dict', '.model', '.png', 'acc_', 'f1_', '.origin', '.adv', '.csv']
-    dataset_file = {'train': [], 'test': [], 'valid': []}
-    search_path = './'
-    task = 'apc_datasets'
-    dataset_file['test'] += find_files(search_path, [dataset, 'test', task, '.inference'], exclude_key=['.adv', '.org', '.defense', 'train.'] + filter_key_words)
+    task = 'apc'
+    dataset_file = detect_infer_dataset(dataset_items[dataset], task)
 
-    for fname in dataset_file['test']:
+    for fname in dataset_file:
         lines = []
         if isinstance(fname, str):
             fname = [fname]