Spaces:
Sleeping
Sleeping
sravanneeli
commited on
Commit
·
9224ca8
1
Parent(s):
44ba809
keep only stable released models
Browse files- vision_models.py +3 -3
vision_models.py
CHANGED
|
@@ -6,9 +6,9 @@ import numpy as np
|
|
| 6 |
classification_models = {
|
| 7 |
"ResNet18": "resnet_18_imagenet",
|
| 8 |
"ResNet50": "resnet_50_imagenet",
|
| 9 |
-
"ViT-B16-224": "vit_base_patch16_224_imagenet",
|
| 10 |
-
"ViT-B16-384": "vit_base_patch16_384_imagenet",
|
| 11 |
-
"ViT-L16-224": "vit_large_patch16_224_imagenet"
|
| 12 |
}
|
| 13 |
|
| 14 |
|
|
|
|
| 6 |
classification_models = {
|
| 7 |
"ResNet18": "resnet_18_imagenet",
|
| 8 |
"ResNet50": "resnet_50_imagenet",
|
| 9 |
+
# "ViT-B16-224": "vit_base_patch16_224_imagenet",
|
| 10 |
+
# "ViT-B16-384": "vit_base_patch16_384_imagenet",
|
| 11 |
+
# "ViT-L16-224": "vit_large_patch16_224_imagenet"
|
| 12 |
}
|
| 13 |
|
| 14 |
|