wufan commited on
Commit
7221ed4
1 Parent(s): 9eb5356

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -13,6 +13,8 @@ from huggingface_hub import snapshot_download
13
  tiny_model_dir = snapshot_download('wanderkid/unimernet_tiny')
14
  small_model_dir = snapshot_download('wanderkid/unimernet_small')
15
  base_model_dir = snapshot_download('wanderkid/unimernet_base')
 
 
16
 
17
  root_path = os.path.abspath(os.getcwd())
18
  os.makedirs(os.path.join(root_path, "models"), exist_ok=True)
 
13
  tiny_model_dir = snapshot_download('wanderkid/unimernet_tiny')
14
  small_model_dir = snapshot_download('wanderkid/unimernet_small')
15
  base_model_dir = snapshot_download('wanderkid/unimernet_base')
16
+ print("download done!")
17
+ print(tiny_model_dir)
18
 
19
  root_path = os.path.abspath(os.getcwd())
20
  os.makedirs(os.path.join(root_path, "models"), exist_ok=True)
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  unimernet==0.1.6
2
- gradio
 
 
1
  unimernet==0.1.6
2
+ gradio
3
+ huggingface_hub