arslanarjumand commited on
Commit
9a1d6c9
1 Parent(s): 0069b4d

Update logic.py

Browse files
Files changed (1) hide show
  1. logic.py +1 -1
logic.py CHANGED
@@ -9,7 +9,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
9
  # os.environ['TRANSFORMERS_VERBOSITY'] = 'error'
10
  torch.random.manual_seed(0);
11
  # protobuf==3.20.0
12
- model_name = 'shawn-nyk/wav2vec2-base-960h-with-lm'
13
 
14
  processor = Wav2Vec2Processor.from_pretrained(model_name)
15
  model = Wav2Vec2ForCTC.from_pretrained(model_name)
 
9
  # os.environ['TRANSFORMERS_VERBOSITY'] = 'error'
10
  torch.random.manual_seed(0);
11
  # protobuf==3.20.0
12
+ model_name = 'jonatasgrosman/wav2vec2-large-xlsr-53-english'
13
 
14
  processor = Wav2Vec2Processor.from_pretrained(model_name)
15
  model = Wav2Vec2ForCTC.from_pretrained(model_name)