Update server.py

server.py

@@ -41,17 +41,25 @@ INITIALIZATION_STATUS = {
 # Global model instance
 model = None
 
+
 def initialize_model():
-    """Initialize the model from mounted directory"""
+    """Initialize the model with correct path resolution"""
     global model, INITIALIZATION_STATUS
     try:
         device = "cuda" if torch.cuda.is_available() else "cpu"
         logger.info(f"Initializing model on device: {device}")
         
-        model_path = os.getenv("MODEL_PATH", "/app/models")
+        # Critical: Use absolute path for model loading
+        model_path = os.path.abspath(os.path.join('/app/src', 'models'))
+        logger.info(f"Loading models from: {model_path}")
+        
         if not os.path.exists(model_path):
             raise RuntimeError(f"Model path {model_path} does not exist")
-
+            
+        # Log available model files for debugging
+        model_files = os.listdir(model_path)
+        logger.info(f"Available model files: {model_files}")
+        
         model = InferenceRecipe(model_path, device=device)
         INITIALIZATION_STATUS["model_loaded"] = True
         logger.info("Model initialized successfully")
@@ -60,7 +68,7 @@ def initialize_model():
         INITIALIZATION_STATUS["error"] = str(e)
         logger.error(f"Failed to initialize model: {e}")
         return False
-
+        
 @app.on_event("startup")
 async def startup_event():
     """Initialize model on startup"""