Vaibhav Srivastav commited on
Commit
851eb15
1 Parent(s): be02097

adding hubert cache

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ nltk.download("punkt")
9
 
10
  wav2vec2processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
11
  wav2vec2model = AutoModelForCTC.from_pretrained("facebook/wav2vec2-base-960h")
 
 
12
 
13
  def return_processor_and_model(model_name):
14
  return Wav2Vec2Processor.from_pretrained(model_name), AutoModelForCTC.from_pretrained(model_name)
 
9
 
10
  wav2vec2processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
11
  wav2vec2model = AutoModelForCTC.from_pretrained("facebook/wav2vec2-base-960h")
12
+ hubertprocessor = Wav2Vec2Processor.from_pretrained("facebook/hubert-large-ls960-ft")
13
+ hubertmodel = AutoModelForCTC.from_pretrained("facebook/hubert-large-ls960-ft")
14
 
15
  def return_processor_and_model(model_name):
16
  return Wav2Vec2Processor.from_pretrained(model_name), AutoModelForCTC.from_pretrained(model_name)