Update app.py

app.py

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