admin commited on
Commit
8becfc8
·
1 Parent(s): 4ad92fc
Files changed (2) hide show
  1. model.py +1 -1
  2. requirements.txt +3 -5
model.py CHANGED
@@ -84,7 +84,7 @@ class EvalNet:
84
  backbone_list = (
85
  load_dataset("monetjoe/cv_backbones", split="train")
86
  if EN_US
87
- else MsDataset.load("monetjoe/cv_backbones", split="v1")
88
  )
89
  backbone_info = self._get_backbone(backbone, backbone_list)
90
  return (
 
84
  backbone_list = (
85
  load_dataset("monetjoe/cv_backbones", split="train")
86
  if EN_US
87
+ else MsDataset.load("monetjoe/cv_backbones", split="train")
88
  )
89
  backbone_info = self._get_backbone(backbone, backbone_list)
90
  return (
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