Update app.py

app.py

@@ -104,11 +104,11 @@ def visualize(dataset_path):
 visualize(DatasetConfig.DATA_ROOT)
 
 train_df = pd.read_fwf(
-    os.path.join(DatasetConfig.TRAIN_DATA_ROOT, 'filetxt/DATASET TXT/train/train.txt'), header=None
+    os.path.join(DatasetConfig.DATA_ROOT, 'filetxt/DATASET TXT/train/train.txt'), header=None
 )
 train_df.rename(columns={0: 'file_name', 1: 'text'}, inplace=True)
 test_df = pd.read_fwf(
-    os.path.join(DatasetConfig.TEST_DATA_ROOT, 'filetxt/DATASET TXT/test/test.txt'), header=None
+    os.path.join(DatasetConfig.DATA_ROOT, 'filetxt/DATASET TXT/test/test.txt'), header=None
 )
 test_df.rename(columns={0: 'file_name', 1: 'text'}, inplace=True)