Update prediction.py

prediction.py

@@ -20,15 +20,17 @@ def preprocess_input_image(img_path):
     x /= 255.
     return x, img1
 
-main_path= '\\food'
     
 # Define batch size and image size
 batch_size = 256
 img_size = (64, 64)
 # Define paths to the data folders
-train_path = '\\food\\Train'
-valid_path = '\\food\\Valid'
-test_path = '\\food\\Test'
+# Define paths to the data folders
+script_dir = os.path.dirname(os.path.abspath(__file__))
+    
+train_path = os.path.join(script_dir, 'food', 'Train')
+valid_path = os.path.join(script_dir, 'food', 'Valid')
+test_path = os.path.join(script_dir, 'food', 'Test')
 
 # Create data generators for training, validation, and testing
 train_datagen = ImageDataGenerator(