andrewsunanda commited on
Commit
d5063eb
·
1 Parent(s): fe46428

Update eda.py

Browse files
Files changed (1) hide show
  1. eda.py +6 -4
eda.py CHANGED
@@ -27,15 +27,17 @@ def run():
27
 
28
 
29
  st.markdown('---')
30
- main_path= '\\food'
31
 
32
  # Define batch size and image size
33
  batch_size = 256
34
  img_size = (64, 64)
35
  # Define paths to the data folders
36
- train_path = '\\food\\Train'
37
- valid_path = '\\food\\Valid'
38
- test_path = '\\food\\Test'
 
 
39
  # Create data generators for training, validation, and testing
40
  train_datagen = ImageDataGenerator(
41
  rescale=1./255,
 
27
 
28
 
29
  st.markdown('---')
30
+
31
 
32
  # Define batch size and image size
33
  batch_size = 256
34
  img_size = (64, 64)
35
  # Define paths to the data folders
36
+ script_dir = os.path.dirname(os.path.abspath(__file__))
37
+
38
+ train_path = os.path.join(script_dir, 'food', 'Train')
39
+ valid_path = os.path.join(script_dir, 'food', 'Valid')
40
+ test_path = os.path.join(script_dir, 'food', 'Test')
41
  # Create data generators for training, validation, and testing
42
  train_datagen = ImageDataGenerator(
43
  rescale=1./255,