datnth1709 commited on
Commit
e6ce204
1 Parent(s): bfcc104

realtime translate

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ def speech2text_vi(audio):
80
  """English speech2text"""
81
  nltk.download("punkt")
82
  # Loading the model and the tokenizer
83
- model_name = "facebook/wav2vec2-base-960h"
84
  eng_tokenizer = Wav2Vec2Tokenizer.from_pretrained(model_name)
85
  eng_model = Wav2Vec2ForCTC.from_pretrained(model_name)
86
 
 
80
  """English speech2text"""
81
  nltk.download("punkt")
82
  # Loading the model and the tokenizer
83
+ model_name = "facebook/s2t-small-librispeech-asr"
84
  eng_tokenizer = Wav2Vec2Tokenizer.from_pretrained(model_name)
85
  eng_model = Wav2Vec2ForCTC.from_pretrained(model_name)
86