arslanarjumand
commited on
Commit
•
9a1d6c9
1
Parent(s):
0069b4d
Update logic.py
Browse files
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 = '
|
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)
|