Hugging Face's logo

Spaces:
SharkSpace
/
shark_detection
Runtime error

App Files Community
Ivan Felipe Rodriguez commited on
Commit
1d1e3d3
·
1 Parent(s): abe0334

removing pdb

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -102,7 +102,7 @@ register_all_modules()
102
  model = init_detector(config_file, checkpoint_file, device='cuda:0') # or device='cuda:0'
103
  model.dataset_meta['classes'] = classes
104
  model.dataset_meta['thing_classes'] = classes
105
- model.dataset_meta['stuff_classes'] = []
106
  model.dataset_meta['palette'] = model.dataset_meta['palette']*3
107
 
108
  print(model.cfg.visualizer)
 
102
  model = init_detector(config_file, checkpoint_file, device='cuda:0') # or device='cuda:0'
103
  model.dataset_meta['classes'] = classes
104
  model.dataset_meta['thing_classes'] = classes
105
+
106
  model.dataset_meta['palette'] = model.dataset_meta['palette']*3
107
 
108
  print(model.cfg.visualizer)