download model online
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ torch.backends.cuda.matmul.allow_tf32 = True
|
|
| 24 |
torch.backends.cudnn.allow_tf32 = True
|
| 25 |
import gc
|
| 26 |
from datetime import datetime
|
| 27 |
-
|
| 28 |
log = logging.getLogger()
|
| 29 |
|
| 30 |
device = "cpu"
|
|
@@ -34,7 +34,7 @@ setup_eval_logging()
|
|
| 34 |
|
| 35 |
OUTPUT_DIR = Path("./output/gradio")
|
| 36 |
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
|
| 37 |
-
|
| 38 |
current_model_state = {
|
| 39 |
"net": None,
|
| 40 |
"feature_utils": None,
|
|
|
|
| 24 |
torch.backends.cudnn.allow_tf32 = True
|
| 25 |
import gc
|
| 26 |
from datetime import datetime
|
| 27 |
+
from huggingface_hub import snapshot_download
|
| 28 |
log = logging.getLogger()
|
| 29 |
|
| 30 |
device = "cpu"
|
|
|
|
| 34 |
|
| 35 |
OUTPUT_DIR = Path("./output/gradio")
|
| 36 |
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
|
| 37 |
+
snapshot_download(repo_id="junxiliu/Meanaudio", local_dir="./weights",allow_patterns=["*.pt", "*.pth"] )
|
| 38 |
current_model_state = {
|
| 39 |
"net": None,
|
| 40 |
"feature_utils": None,
|