Update prediction.py

prediction.py

@@ -23,17 +23,19 @@ def preprocess_input_image(img_path):
     x /= 255.
     return x, img1
 
-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(