Update app.py

app.py

@@ -103,11 +103,11 @@ def visualize(dataset_path):
 visualize(DatasetConfig.DATA_ROOT)
 
 train_df = pd.read_fwf(
-    os.path.join(DatasetConfig.DATA_ROOT, '/content/DATASET TXT/train/train.txt'), header=None
+    os.path.join(DatasetConfig.DATA_ROOT, '/main/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, '/content/DATASET TXT/test/test.txt'), header=None
+    os.path.join(DatasetConfig.DATA_ROOT, '/main/DATASET TXT/test/test.txt'), header=None
 )
 test_df.rename(columns={0: 'file_name', 1: 'text'}, inplace=True)