Jihuai commited on
Commit
c2f3ea7
·
1 Parent(s): f34b3d3

edit app for config_root env var

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -9,6 +9,7 @@ import tempfile
9
  USE_CUDA = torch.cuda.is_available()
10
  BATCH_SIZE = int(os.getenv("BATCH_SIZE", "12"))
11
 
 
12
  # Read model repo and filename from environment variables
13
  REPO_ID = os.getenv("MODEL_REPO_ID", "chenxie95/Language-Audio-Banquet-ckpt")
14
  FILENAME = os.getenv("MODEL_FILENAME", "ev-pre-aug.ckpt")
 
9
  USE_CUDA = torch.cuda.is_available()
10
  BATCH_SIZE = int(os.getenv("BATCH_SIZE", "12"))
11
 
12
+ os.environ['CONFIG_ROOT'] = './config'
13
  # Read model repo and filename from environment variables
14
  REPO_ID = os.getenv("MODEL_REPO_ID", "chenxie95/Language-Audio-Banquet-ckpt")
15
  FILENAME = os.getenv("MODEL_FILENAME", "ev-pre-aug.ckpt")