Update the HF model URL
Browse files- inference.py +2 -2
inference.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
# Copyright © 2024 Roblox Corporation
|
2 |
|
3 |
"""
|
4 |
-
This file gives a sample demonstration of how to use the given functions in Python, for the Voice Safety Classifier model.
|
5 |
"""
|
6 |
|
7 |
import torch
|
@@ -82,7 +82,7 @@ if __name__ == "__main__":
|
|
82 |
parser.add_argument(
|
83 |
"--model_path",
|
84 |
type=str,
|
85 |
-
default="roblox/voice-safety-classifier",
|
86 |
help="checkpoint file of model",
|
87 |
)
|
88 |
args = parser.parse_args()
|
|
|
1 |
# Copyright © 2024 Roblox Corporation
|
2 |
|
3 |
"""
|
4 |
+
This file gives a sample demonstration of how to use the given functions in Python, for the Voice Safety Classifier v2 model.
|
5 |
"""
|
6 |
|
7 |
import torch
|
|
|
82 |
parser.add_argument(
|
83 |
"--model_path",
|
84 |
type=str,
|
85 |
+
default="roblox/voice-safety-classifier-v2",
|
86 |
help="checkpoint file of model",
|
87 |
)
|
88 |
args = parser.parse_args()
|