Minor changes

main/app.py

@@ -4,11 +4,8 @@ from fastapi.middleware.cors import CORSMiddleware
 from .routes import router, init_router
 from .utils.logging import setup_logger
 from .utils.validation import validate_hf
+from .utils.helpers import load_config
 
-def load_config():
-    """Load configuration from yaml file"""
-    with open("main/config.yaml", "r") as f:
-        return yaml.safe_load(f)
 
 config = load_config()
 logger = setup_logger(config, "main")

main/utils/helpers.py

@@ -3,6 +3,9 @@ import torch
 from pathlib import Path
 from typing import Dict, Any
 
+import yaml
+
+
 def get_system_info() -> Dict[str, Any]:
     """Get system resource information"""
     return {
@@ -33,4 +36,9 @@ def format_memory_size(size_bytes: int) -> str:
     for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
         if size_bytes < 1024:
             return f"{size_bytes:.2f}{unit}"
-        size_bytes /= 1024
+        size_bytes /= 1024
+
+def load_config():
+    """Load configuration from yaml file"""
+    with open("main/resources/config.yaml", "r") as f:
+        return yaml.safe_load(f)