PB Unity commited on
Commit
3ce6f24
·
verified ·
1 Parent(s): 0468533

Upload RunWhisper.cs

Browse files
Files changed (1) hide show
  1. RunWhisper.cs +15 -5
RunWhisper.cs CHANGED
@@ -39,12 +39,17 @@ public class RunWhisper : MonoBehaviour
39
  // This is how many tokens you want. It can be adjusted.
40
  const int maxTokens = 100;
41
 
42
- //Special tokens
43
  const int END_OF_TEXT = 50257;
44
  const int START_OF_TRANSCRIPT = 50258;
45
  const int ENGLISH = 50259;
46
- const int TRANSCRIBE = 50359;
 
 
 
 
47
  const int START_TIME = 50364;
 
48
 
49
  Ops ops;
50
  ITensorAllocator allocator;
@@ -85,9 +90,9 @@ public class RunWhisper : MonoBehaviour
85
  spectroEngine = WorkerFactory.CreateWorker(backend, spectro);
86
 
87
  outputTokens[0] = START_OF_TRANSCRIPT;
88
- outputTokens[1] = ENGLISH;
89
- outputTokens[2] = TRANSCRIBE;
90
- outputTokens[3] = START_TIME;
91
  currentToken = 3;
92
 
93
  LoadAudio();
@@ -210,6 +215,11 @@ public class RunWhisper : MonoBehaviour
210
  return !(('!' <= c && c <= '~') || ('�' <= c && c <= '�') || ('�' <= c && c <= '�'));
211
  }
212
 
 
 
 
 
 
213
  private void OnDestroy()
214
  {
215
  decoderEngine?.Dispose();
 
39
  // This is how many tokens you want. It can be adjusted.
40
  const int maxTokens = 100;
41
 
42
+ //Special tokens see added tokens file for details
43
  const int END_OF_TEXT = 50257;
44
  const int START_OF_TRANSCRIPT = 50258;
45
  const int ENGLISH = 50259;
46
+ const int GERMAN = 50261;
47
+ const int FRENCH = 50265;
48
+ const int TRANSCRIBE = 50359; //for speech-to-text in specified language
49
+ const int TRANSLATE = 50358; //for speech-to-text then translate to English
50
+ const int NO_TIME_STAMPS = 50363;
51
  const int START_TIME = 50364;
52
+
53
 
54
  Ops ops;
55
  ITensorAllocator allocator;
 
90
  spectroEngine = WorkerFactory.CreateWorker(backend, spectro);
91
 
92
  outputTokens[0] = START_OF_TRANSCRIPT;
93
+ outputTokens[1] = ENGLISH;// GERMAN;//FRENCH;//
94
+ outputTokens[2] = TRANSCRIBE; //TRANSLATE;//
95
+ outputTokens[3] = NO_TIME_STAMPS;// START_TIME;//
96
  currentToken = 3;
97
 
98
  LoadAudio();
 
215
  return !(('!' <= c && c <= '~') || ('�' <= c && c <= '�') || ('�' <= c && c <= '�'));
216
  }
217
 
218
+ private void OnApplicationQuit()
219
+ {
220
+ if (Input.GetKeyDown(KeyCode.Escape)) Application.Quit();
221
+ }
222
+
223
  private void OnDestroy()
224
  {
225
  decoderEngine?.Dispose();