Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import sys
 from pathlib import Path
 
 from huggingface_hub import snapshot_download
@@ -10,6 +11,11 @@ from huggingface_hub import snapshot_download
 # Files will be placed under ckpt/ so the existing config paths keep working.
 # ---------------------------------------------------------------------------
 BASE_DIR = Path(__file__).parent.resolve()
+ROOT_DIR = BASE_DIR.parent.resolve()
+
+# Ensure project root is on sys.path so intra-repo imports (e.g. `optimizers`) work.
+if str(ROOT_DIR) not in sys.path:
+    sys.path.insert(0, str(ROOT_DIR))
 CKPT_DIR = BASE_DIR / "ckpt"
 CKPT_DIR.mkdir(parents=True, exist_ok=True)