Update app.py

app.py

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