Skip to content

Commit 3335556

Browse files
committed
Update the HF model URL
1 parent ddccf43 commit 3335556

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -51,5 +51,5 @@ To run the inference, please run the following command:
5151
python inference.py --audio_file <your audio file path> --model_path <path to Huggingface model>
5252
```
5353
You can download the model weights from the model releases page [here](https://github.com/Roblox/voice-safety-classifier/releases/tag/vs-classifier-v2),
54-
or from HuggingFace under [`roblox/voice-safety-classifier`](https://huggingface.co/Roblox/voice-safety-classifier).
54+
or from HuggingFace under [`roblox/voice-safety-classifier-v2`](https://huggingface.co/Roblox/voice-safety-classifier-v2).
5555
If `model_path` isn’t specified, the model will be loaded directly from HuggingFace.

inference.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Copyright © 2024 Roblox Corporation
22

33
"""
4-
This file gives a sample demonstration of how to use the given functions in Python, for the Voice Safety Classifier model.
4+
This file gives a sample demonstration of how to use the given functions in Python, for the Voice Safety Classifier v2 model.
55
"""
66

77
import torch
@@ -82,7 +82,7 @@ def infer(model, inputs):
8282
parser.add_argument(
8383
"--model_path",
8484
type=str,
85-
default="roblox/voice-safety-classifier",
85+
default="roblox/voice-safety-classifier-v2",
8686
help="checkpoint file of model",
8787
)
8888
args = parser.parse_args()

0 commit comments

Comments
 (0)