fcakyon commited on
Commit
2ad9d60
1 Parent(s): 7b90c9c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,8 +47,8 @@ def get_model():
47
 
48
  detection_model = AutoDetectionModel.from_pretrained(
49
  model_type='mmdet',
50
- model_path=model_path,
51
- config_path=config_path,
52
  confidence_threshold=0.5,
53
  device="cpu",
54
  )
47
 
48
  detection_model = AutoDetectionModel.from_pretrained(
49
  model_type='mmdet',
50
+ model_path=MMDET_YOLOX_TINY_MODEL_PATH,
51
+ config_path=MMDET_YOLOX_TINY_CONFIG_PATH,
52
  confidence_threshold=0.5,
53
  device="cpu",
54
  )