moro23 commited on
Commit
1da4128
·
1 Parent(s): 900aeb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,12 +7,12 @@ from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
7
  nltk.download("punkt")
8
 
9
  ##
10
- #token_value = "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"
11
  #Loading the pre-trained model and the tokenizer
12
  #model_name = "moro23/wav2vec-large-xls-r-300-ha-colab_4"
13
  #tokenizer = Wav2Vec2Tokenizer.from_pretrained(model_name, use_auth_token=token_value)
14
- #tokenizer = Wav2Vec2Processor.from_pretrained(model_name, use_auth_token=token_value)
15
- #model = Wav2Vec2ForCTC.from_pretrained(model_name, use_auth_token=token_value).to("cuda")
16
 
17
  def load_data(input_file):
18
 
 
7
  nltk.download("punkt")
8
 
9
  ##
10
+ token_value = "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"
11
  #Loading the pre-trained model and the tokenizer
12
  #model_name = "moro23/wav2vec-large-xls-r-300-ha-colab_4"
13
  #tokenizer = Wav2Vec2Tokenizer.from_pretrained(model_name, use_auth_token=token_value)
14
+ tokenizer = Wav2Vec2Processor.from_pretrained(model_name, use_auth_token=token_value)
15
+ model = Wav2Vec2ForCTC.from_pretrained(model_name, use_auth_token=token_value).to("cuda")
16
 
17
  def load_data(input_file):
18