Update eda.py

eda.py

@@ -7,6 +7,9 @@ import matplotlib.pyplot as plt
 import plotly.express as px
 from tensorflow.keras.preprocessing.image import ImageDataGenerator
 from tensorflow.keras.preprocessing import image
+from datasets import load_dataset
+from torchvision import transforms
+from torch.utils.data import DataLoader
 
 st.set_page_config(page_title='Fast Food Classification Dataset Analysis', layout='wide', initial_sidebar_state='expanded')
 
@@ -27,15 +30,15 @@ def run():
     
 
     st.markdown('---')
-    main_path= 'D:\\tugas_andrew_DS\\phase_2\\m2\\food'
+    dataset = load_dataset('andrewsunanda/fast_food_image_classification')
     
     # Define batch size and image size
     batch_size = 256
     img_size = (64, 64)
     # Define paths to the data folders
-    train_path = os.path.join(main_path, 'Train')
-    valid_path = os.path.join(main_path, 'Valid')
-    test_path = os.path.join(main_path, 'Test')
+    train_path = dataset['train'].features['filepath']
+    valid_path = dataset['validation'].features['filepath']
+    test_path = dataset['test'].features['filepath']
     # Create data generators for training, validation, and testing
     train_datagen = ImageDataGenerator(
         rescale=1./255,