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()
|