qubvel-hf HF Staff commited on
Commit
eff589b
·
verified ·
1 Parent(s): 83610db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -22,19 +22,19 @@ from transformers.image_utils import load_image
22
 
23
  # Configuration constants
24
  CHECKPOINTS = [
25
- "ustc-community/dfine_m_obj2coco",
26
- "ustc-community/dfine_m_obj365",
27
- "ustc-community/dfine_n_coco",
28
- "ustc-community/dfine_s_coco",
29
- "ustc-community/dfine_m_coco",
30
- "ustc-community/dfine_l_coco",
31
- "ustc-community/dfine_x_coco",
32
- "ustc-community/dfine_s_obj365",
33
- "ustc-community/dfine_l_obj365",
34
- "ustc-community/dfine_x_obj365",
35
- "ustc-community/dfine_s_obj2coco",
36
- "ustc-community/dfine_l_obj2coco_e25",
37
- "ustc-community/dfine_x_obj2coco",
38
  ]
39
  DEFAULT_CHECKPOINT = CHECKPOINTS[0]
40
  DEFAULT_CONFIDENCE_THRESHOLD = 0.3
 
22
 
23
  # Configuration constants
24
  CHECKPOINTS = [
25
+ "ustc-community/dfine-medium-obj2coco",
26
+ "ustc-community/dfine-medium-coco",
27
+ "ustc-community/dfine-medium-obj365",
28
+ "ustc-community/dfine-nano-coco",
29
+ "ustc-community/dfine-small-coco",
30
+ "ustc-community/dfine-large-coco",
31
+ "ustc-community/dfine-xlarge-coco",
32
+ "ustc-community/dfine-small-obj365",
33
+ "ustc-community/dfine-large-obj365",
34
+ "ustc-community/dfine-xlarge-obj365",
35
+ "ustc-community/dfine-small-obj2coco",
36
+ "ustc-community/dfine-large-obj2coco-e25",
37
+ "ustc-community/dfine-xlarge-obj2coco",
38
  ]
39
  DEFAULT_CHECKPOINT = CHECKPOINTS[0]
40
  DEFAULT_CONFIDENCE_THRESHOLD = 0.3