Spaces:
Running
Running
benjamin-paine
commited on
Commit
•
d444944
1
Parent(s):
39d4b05
Update index.js
Browse files
index.js
CHANGED
@@ -10,12 +10,16 @@ const colors = {
|
|
10 |
"frame budget": [25,255,25]
|
11 |
};
|
12 |
|
|
|
13 |
const wakeWords = ["buddy", "hey buddy", "hi buddy", "sup buddy", "yo buddy", "okay buddy"];
|
14 |
const canvasSize = { width: 640, height: 100 };
|
15 |
const graphLineWidth = 1;
|
16 |
const options = {
|
17 |
debug: true,
|
18 |
-
modelPath: wakeWords.map((word) =>
|
|
|
|
|
|
|
19 |
};
|
20 |
|
21 |
/** Helper method for conversion */
|
|
|
10 |
"frame budget": [25,255,25]
|
11 |
};
|
12 |
|
13 |
+
const rootUrl = "https://huggingface.co/benjamin-paine/hey-buddy/resolve/main";
|
14 |
const wakeWords = ["buddy", "hey buddy", "hi buddy", "sup buddy", "yo buddy", "okay buddy"];
|
15 |
const canvasSize = { width: 640, height: 100 };
|
16 |
const graphLineWidth = 1;
|
17 |
const options = {
|
18 |
debug: true,
|
19 |
+
modelPath: wakeWords.map((word) => `${rootUrl}/models/${word.replace(' ', '-')}.onnx`),
|
20 |
+
vadModelPath: `${rootUrl}/pretrained/silero-vad.onnx`,
|
21 |
+
spectrogramModelPath: `${rootUrl}/pretrained/mel-spectrogram.onnx`,
|
22 |
+
embeddingModelPath: `${rootUrl}/pretrained/speech-embedding.onnx`,
|
23 |
};
|
24 |
|
25 |
/** Helper method for conversion */
|