Vaibhav Srivastav commited on
Commit
1058cee
·
1 Parent(s): b533f0d

reverting changes

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,10 +7,10 @@ from transformers import Wav2Vec2Processor, AutoModelForCTC
7
 
8
  nltk.download("punkt")
9
 
10
- # wav2vec2_processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
11
- # wav2vec2_model = AutoModelForCTC.from_pretrained("facebook/wav2vec2-base-960h")
12
- # hubert_processor = Wav2Vec2Processor.from_pretrained("facebook/hubert-large-ls960-ft")
13
- # hubert_model = AutoModelForCTC.from_pretrained("facebook/hubert-large-ls960-ft")
14
 
15
  def return_processor_and_model(model_name):
16
  if model_name == "facebook/wav2vec2-base-960h":
 
7
 
8
  nltk.download("punkt")
9
 
10
+ wav2vec2_processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
11
+ wav2vec2_model = AutoModelForCTC.from_pretrained("facebook/wav2vec2-base-960h")
12
+ hubert_processor = Wav2Vec2Processor.from_pretrained("facebook/hubert-large-ls960-ft")
13
+ hubert_model = AutoModelForCTC.from_pretrained("facebook/hubert-large-ls960-ft")
14
 
15
  def return_processor_and_model(model_name):
16
  if model_name == "facebook/wav2vec2-base-960h":