admin commited on
Commit
e7824ef
·
1 Parent(s): 87f3877

huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
  from PIL import Image
7
  from model import Model
8
  from torchvision import transforms
9
- from modelscope import snapshot_download
10
 
11
 
12
  MODEL_DIR = snapshot_download("Genius-Society/svhn", cache_dir="./__pycache__")
 
6
  from PIL import Image
7
  from model import Model
8
  from torchvision import transforms
9
+ from huggingface_hub import snapshot_download
10
 
11
 
12
  MODEL_DIR = snapshot_download("Genius-Society/svhn", cache_dir="./__pycache__")
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
  gradio
2
  pillow
3
  torch
4
- torchvision
5
- modelscope==1.15
 
1
  gradio
2
  pillow
3
  torch
4
+ torchvision