Spaces:
Running
Running
MuGeminorum
commited on
Commit
•
ec9cfb0
1
Parent(s):
24beb0a
try to fix runtime err
Browse files- model.py +13 -1
- 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
|