Fix config loading bug

app.py

@@ -25,8 +25,8 @@ def load_config(config_path):
 def load_model(dataset_name):
     config = load_config(CONFIG_PATHS[dataset_name])
     model = UNet(
-        in_channels=config['model']['in_channels'],
-        out_channels=config['model']['out_channels']
+        in_channels=config['model']['params']['in_channels'],
+        out_channels=config['model']['params']['out_channels']
     )
     model_path = MODEL_PATHS[dataset_name]
     if os.path.exists(model_path):