fix

app.py

@@ -680,7 +680,7 @@ models = {}
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load Hawk model
-hawk_config = load_model_config("hawk")
+hawk_config = load_model_config("hawk", models_dir="deployment/models")
 input_size_hawk = get_input_size(hawk_config)
 hawk_model = HawkPredictor(
     input_size=input_size_hawk,
@@ -695,7 +695,7 @@ hawk_model.eval()
 models["hawk"] = hawk_model
 
 # Load Mamba2 model
-mamba_config = load_model_config("mamba")
+mamba_config = load_model_config("mamba", models_dir="deployment/models")
 input_size_mamba = get_input_size(mamba_config)
 mamba_model = Mamba2Predictor(
     input_size=input_size_mamba,
@@ -713,7 +713,7 @@ mamba_model.eval()
 models["mamba"] = mamba_model
 
 # Load xLSTM model
-xlstm_config = load_model_config("xlstm")
+xlstm_config = load_model_config("xlstm", models_dir="deployment/models")
 input_size_xlstm = get_input_size(xlstm_config)
 xlstm_model = xLSTMPredictor(
     input_size=input_size_xlstm,