Update app.py

app.py

@@ -29,8 +29,8 @@ config = Config.fromfile(config_path)
 config.model.backbone.pretrained = None
 model = init_segmentor(config, ckpt, device='cpu')
 
-# Define the preprocessing pipeline
-custom_test_pipeline = process_test_pipeline(model.cfg.data.test.pipeline, None)
+# Use the test pipeline directly
+custom_test_pipeline = model.cfg.data.test.pipeline
 
 # Define the input/output model for FastAPI
 class PredictionOutput(BaseModel):