benjamin-paine commited on
Commit
d444944
1 Parent(s): 39d4b05

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +5 -1
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) => `https://huggingface.co/benjamin-paine/hey-buddy/resolve/main/models/${word.replace(' ', '-')}.onnx`)
 
 
 
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 */