sanket09 commited on
Commit
06355df
·
verified ·
1 Parent(s): 2b313f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ config = Config.fromfile(config_path)
29
  config.model.backbone.pretrained = None
30
  model = init_segmentor(config, ckpt, device='cpu')
31
 
32
- # Define the preprocessing pipeline
33
- custom_test_pipeline = process_test_pipeline(model.cfg.data.test.pipeline, None)
34
 
35
  # Define the input/output model for FastAPI
36
  class PredictionOutput(BaseModel):
 
29
  config.model.backbone.pretrained = None
30
  model = init_segmentor(config, ckpt, device='cpu')
31
 
32
+ # Use the test pipeline directly
33
+ custom_test_pipeline = model.cfg.data.test.pipeline
34
 
35
  # Define the input/output model for FastAPI
36
  class PredictionOutput(BaseModel):