renyuxi commited on
Commit
da1d919
1 Parent(s): 0352a11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import os
3
  import time
4
  from os import path
5
  from safetensors.torch import load_file
 
6
 
7
  cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
8
  os.environ["TRANSFORMERS_CACHE"] = cache_path
 
3
  import time
4
  from os import path
5
  from safetensors.torch import load_file
6
+ from huggingface_hub import hf_hub_download
7
 
8
  cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
9
  os.environ["TRANSFORMERS_CACHE"] = cache_path