admin commited on
Commit
c229fbb
·
1 Parent(s): b152d6c
Files changed (2) hide show
  1. model.py +1 -1
  2. requirements.txt +3 -5
model.py CHANGED
@@ -42,7 +42,7 @@ class EvalNet:
42
  backbone_list = (
43
  load_dataset("monetjoe/cv_backbones", split="train")
44
  if EN_US
45
- else MsDataset.load("monetjoe/cv_backbones", split="v1")
46
  )
47
  backbone = self._get_backbone(m_ver, backbone_list)
48
  m_type = str(backbone["type"])
 
42
  backbone_list = (
43
  load_dataset("monetjoe/cv_backbones", split="train")
44
  if EN_US
45
+ else MsDataset.load("monetjoe/cv_backbones", split="train")
46
  )
47
  backbone = self._get_backbone(m_ver, backbone_list)
48
  m_type = str(backbone["type"])
requirements.txt CHANGED
@@ -1,7 +1,5 @@
1
- torch==2.6.0+cu118
2
- -f https://download.pytorch.org/whl/torch
3
- torchvision==0.21.0+cu118
4
- -f https://download.pytorch.org/whl/torchvision
5
  librosa
6
  matplotlib
7
- modelscope[framework]==1.21.0
 
 
 
 
 
 
 
1
  librosa
2
  matplotlib
3
+ modelscope[framework]
4
+ torch
5
+ torchvision