Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,19 +22,19 @@ from transformers.image_utils import load_image
|
|
| 22 |
|
| 23 |
# Configuration constants
|
| 24 |
CHECKPOINTS = [
|
| 25 |
-
"ustc-community/
|
| 26 |
-
"ustc-community/
|
| 27 |
-
"ustc-community/
|
| 28 |
-
"ustc-community/
|
| 29 |
-
"ustc-community/
|
| 30 |
-
"ustc-community/
|
| 31 |
-
"ustc-community/
|
| 32 |
-
"ustc-community/
|
| 33 |
-
"ustc-community/
|
| 34 |
-
"ustc-community/
|
| 35 |
-
"ustc-community/
|
| 36 |
-
"ustc-community/
|
| 37 |
-
"ustc-community/
|
| 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
|