MuGeminorum commited on
Commit
ec9cfb0
1 Parent(s): 24beb0a

try to fix runtime err

Browse files
Files changed (2) hide show
  1. model.py +13 -1
  2. requirements.txt +0 -2
model.py CHANGED
@@ -1,8 +1,8 @@
1
  import os
 
2
  import torch
3
  import torch.nn as nn
4
  import torchvision.models as models
5
- from modelscope.msdatasets import MsDataset
6
  from datasets import load_dataset
7
  from utils import url_download, create_dir, DOMAIN, DOMAIN_ZH
8
 
@@ -24,6 +24,18 @@ def model_info(m_ver):
24
  trust_remote_code=True
25
  )
26
  except ConnectionError:
 
 
 
 
 
 
 
 
 
 
 
 
27
  backbone_list = MsDataset.load(
28
  'monetjoe/cv_backbones',
29
  subset_name='ImageNet1k_v1',
 
1
  import os
2
+ import subprocess
3
  import torch
4
  import torch.nn as nn
5
  import torchvision.models as models
 
6
  from datasets import load_dataset
7
  from utils import url_download, create_dir, DOMAIN, DOMAIN_ZH
8
 
 
24
  trust_remote_code=True
25
  )
26
  except ConnectionError:
27
+ def is_package_installed(package_name):
28
+ import importlib.metadata
29
+ try:
30
+ importlib.metadata.version(package_name)
31
+ return True
32
+ except importlib.metadata.PackageNotFoundError:
33
+ return False
34
+
35
+ if not is_package_installed('modelscope'):
36
+ subprocess.call('pip install modelscope')
37
+
38
+ from modelscope.msdatasets import MsDataset
39
  backbone_list = MsDataset.load(
40
  'monetjoe/cv_backbones',
41
  subset_name='ImageNet1k_v1',
requirements.txt CHANGED
@@ -4,6 +4,4 @@ matplotlib
4
  torchvision
5
  pillow
6
  gradio
7
- modelscope
8
- datasets
9
  bs4
 
4
  torchvision
5
  pillow
6
  gradio
 
 
7
  bs4