Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ if not torch.cuda.is_available():
|
|
34 |
|
35 |
predictor = DefaultPredictor(cfg)
|
36 |
my_metadata = MetadataCatalog.get("car_parts_dataset_val")
|
37 |
-
my_metadata
|
38 |
'back_bumper',
|
39 |
'back_glass',
|
40 |
'back_left_door',
|
|
|
34 |
|
35 |
predictor = DefaultPredictor(cfg)
|
36 |
my_metadata = MetadataCatalog.get("car_parts_dataset_val")
|
37 |
+
my_metadata.thing_classes = ['_background_',
|
38 |
'back_bumper',
|
39 |
'back_glass',
|
40 |
'back_left_door',
|