Update Training/Code/train.py

Training/Code/train.py

@@ -10,8 +10,10 @@ from tensorflow.keras.optimizers import Adam
 from sklearn.utils.class_weight import compute_class_weight
 
 # ==================== Paths ====================
-train_dir = "/content/combine_dataset/train"
-val_dir = "/content/combine_dataset/test"
+# Define paths
+base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '../'))
+train_dir = os.path.join(base_dir, 'combine_dataset/train')
+val_dir = os.path.join(base_dir, 'combine_dataset/test')
 
 # ==================== Parameters ====================
 img_size = (192, 192)  # Recommended for EfficientNetV2B1