Spaces:

vankienemk
/

Voice-regconizer

Running

vankienemk commited on Apr 10

Commit

1d9d391

verified ·

1 Parent(s): f8da254

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import numpy as np
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
 # Load model
-processor = Wav2Vec2Processor.from_pretrained("FPTAI/vietnamese-wav2vec2-base")
-model = Wav2Vec2ForCTC.from_pretrained("FPTAI/vietnamese-wav2vec2-base")
 def transcribe(audio):
     if audio is None:

 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
 # Load model
+processor = Wav2Vec2Processor.from_pretrained("nguyenvulebinh/wav2vec2-base-vietnamese-250h")
+model = Wav2Vec2ForCTC.from_pretrained("nguyenvulebinh/wav2vec2-base-vietnamese-250h")
 def transcribe(audio):
     if audio is None: