Spaces:
Running
Running
admin
commited on
Commit
·
38b567b
1
Parent(s):
6a186a7
trust_remote_code=True,
Browse files
model.py
CHANGED
|
@@ -38,7 +38,11 @@ class EvalNet:
|
|
| 38 |
return backbone_list[0]
|
| 39 |
|
| 40 |
def _model_info(self, m_ver: str):
|
| 41 |
-
backbone_list = MsDataset.load(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
backbone = self._get_backbone(m_ver, backbone_list)
|
| 43 |
m_type = str(backbone["type"])
|
| 44 |
input_size = int(backbone["input_size"])
|
|
|
|
| 38 |
return backbone_list[0]
|
| 39 |
|
| 40 |
def _model_info(self, m_ver: str):
|
| 41 |
+
backbone_list = MsDataset.load(
|
| 42 |
+
"monetjoe/cv_backbones",
|
| 43 |
+
split="v1",
|
| 44 |
+
trust_remote_code=True,
|
| 45 |
+
)
|
| 46 |
backbone = self._get_backbone(m_ver, backbone_list)
|
| 47 |
m_type = str(backbone["type"])
|
| 48 |
input_size = int(backbone["input_size"])
|