Some modifications

main/main.py

@@ -8,7 +8,6 @@ from pathlib import Path
 from litgpt.api import LLM
 import os
 import uvicorn
-from huggingface_hub.hf_api import HfApi
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -75,7 +74,7 @@ async def initialize_model(request: InitializeRequest):
 
     try:
         # Get the project root directory (where main.py is located)
-        project_root = Path(__file__).parent
+        project_root = Path(__file__).parent.parent
         checkpoints_dir = project_root / "checkpoints"
 
         # For LitGPT downloaded models, path includes organization