Spaces:

ruiite
/

car_damage_detection

Runtime error

ruiite commited on Oct 5, 2022

Commit

84cedfd

•

1 Parent(s): 68ef79c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,6 +18,7 @@ from detectron2.engine import DefaultPredictor
 from detectron2.config import get_cfg
 from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog
 model_path = 'model_final.pth'

 from detectron2.config import get_cfg
 from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog
+from detectron2.utils.visualizer import ColorMode
 model_path = 'model_final.pth'