vineelpratap commited on
Commit
175fc0f
1 Parent(s): 56eee40

Update zeroshot.py

Browse files
Files changed (1) hide show
  1. zeroshot.py +5 -2
zeroshot.py CHANGED
@@ -22,12 +22,15 @@ WORD_SCORE_DEFAULT_IF_LM = -0.18
22
  WORD_SCORE_DEFAULT_IF_NOLM = -3.5
23
  LM_SCORE_DEFAULT = 1.48
24
 
25
- MODEL_ID = "upload/mms_zs"
26
 
27
  processor = AutoProcessor.from_pretrained(MODEL_ID)
28
  model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
29
 
30
- token_file = "upload/mms_zs/tokens.txt"
 
 
 
31
 
32
 
33
  class MY_LOG:
 
22
  WORD_SCORE_DEFAULT_IF_NOLM = -3.5
23
  LM_SCORE_DEFAULT = 1.48
24
 
25
+ MODEL_ID = "mms-meta/mms-zeroshot-300m"
26
 
27
  processor = AutoProcessor.from_pretrained(MODEL_ID)
28
  model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
29
 
30
+ token_file = hf_hub_download(
31
+ repo_id=MODEL_ID,
32
+ filename="tokens.txt",
33
+ )
34
 
35
 
36
  class MY_LOG: