Update Training script

train.py

@@ -13,7 +13,7 @@ CLASSES = ["AMD","DMO","DR","GLC","HR","Normal"]
 MODEL_NAME = "microsoft/beit-base-patch16-224"
 
 # Output directory (from env or default)
-OUTPUT_DIR = os.environ.get("OUTPUT_DIR", os.path.join(os.environ["HOME"], "outputs/beit-retina"))
+OUTPUT_DIR = os.environ.get("OUTPUT_DIR", "/home/user/outputs/beit-retina")
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 print("HOME dir:", os.environ.get("HOME"))