Spaces:
Runtime error
Runtime error
edit app for config_root env var
Browse files
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")
|