Spaces:
Running
Running
admin
commited on
Commit
•
dfca8bb
1
Parent(s):
c893964
fix 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"])
|