Update eda.py

eda.py

@@ -30,17 +30,20 @@ def run():
     
 
     st.markdown('---')
-    dataset_path = "andrewsunanda/fast_food_image_classification"
+    
     
     
     # Define batch size and image size
     batch_size = 256
     img_size = (64, 64)
     # Define paths to the data folders
-    dataset = load_dataset(dataset_path)
-    train_path = dataset['train'].cache_files[0].split('train')[0] + 'train'
-    valid_path = dataset['valid'].cache_files[0].split('valid')[0] + 'valid'
-    test_path = dataset['test'].cache_files[0].split('test')[0] + 'test'
+    dataset_path = 'andrewsunanda/fast_food_image_classification'
+    
+   
+    # Define the paths to the train, validation, and test folders
+    train_path = os.path.join(dataset_path, 'Train')
+    valid_path = os.path.join(dataset_path, 'Valid')
+    test_path = os.path.join(dataset_path, 'Test')
     # Create data generators for training, validation, and testing
     train_datagen = ImageDataGenerator(
         rescale=1./255,