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

Update prediction.py

Browse files
Files changed (1) hide show
  1. prediction.py +3 -3
prediction.py CHANGED
@@ -26,9 +26,9 @@ main_path= '\\food'
26
  batch_size = 256
27
  img_size = (64, 64)
28
  # Define paths to the data folders
29
- train_path = os.path.join(main_path, 'Train')
30
- valid_path = os.path.join(main_path, 'Valid')
31
- test_path = os.path.join(main_path, 'Test')
32
 
33
  # Create data generators for training, validation, and testing
34
  train_datagen = ImageDataGenerator(
 
26
  batch_size = 256
27
  img_size = (64, 64)
28
  # Define paths to the data folders
29
+ train_path = '\\food\\Train'
30
+ valid_path = '\\food\\Valid'
31
+ test_path = '\\food\\Test'
32
 
33
  # Create data generators for training, validation, and testing
34
  train_datagen = ImageDataGenerator(