Update app.py

app.py

@@ -83,6 +83,7 @@ class TrainingConfig:
 @dataclass(frozen=True)
 class DatasetConfig:
     DATA_ROOT:     str = 'DATASET-TRAIN-TEST-1'
+    TRAIN_DATA_ROOT:     str = 'DATASET TXT/train/train.txt'
 
 @dataclass(frozen=True)
 class ModelConfig:
@@ -103,11 +104,11 @@ def visualize(dataset_path):
 visualize(DatasetConfig.DATA_ROOT)
 
 train_df = pd.read_fwf(
-    os.path.join(DatasetConfig.DATA_ROOT, '/DATASET TXT/train/train.txt'), header=None
+    os.path.join(DatasetConfig.TRAIN_DATA_ROOT, '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.DATA_ROOT, '/DATASET TXT/test/test.txt'), header=None
+    os.path.join(DatasetConfig.TRAIN_DATA_ROOT, 'DATASET TXT/test/test.txt'), header=None
 )
 test_df.rename(columns={0: 'file_name', 1: 'text'}, inplace=True)