Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
#
|
33 |
-
custom_test_pipeline =
|
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):
|