ak3ra commited on
Commit
cf4ccb1
·
1 Parent(s): 2c72d05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from transformers import Wav2Vec2ForCTC, AutoProcessor
3
  import torch
4
  import librosa
5
  import json
@@ -13,7 +13,8 @@ auth_token = os.environ.get("HF_TOKEN")
13
 
14
  model_id = "Sunbird/sunbird-mms"
15
  model = Wav2Vec2ForCTC.from_pretrained(model_id, use_auth_token=auth_token)
16
- processor = AutoProcessor.from_pretrained(model_id, use_auth_token=auth_token)
 
17
 
18
  def transcribe(audio_file_mic=None, audio_file_upload=None, language="Luganda (lug)"):
19
  if audio_file_mic:
 
1
  import gradio as gr
2
+ from transformers import Wav2Vec2ForCTC, AutoProcessor, Wav2Vec2Processor
3
  import torch
4
  import librosa
5
  import json
 
13
 
14
  model_id = "Sunbird/sunbird-mms"
15
  model = Wav2Vec2ForCTC.from_pretrained(model_id, use_auth_token=auth_token)
16
+ processor = Wav2Vec2Processor.from_pretrained(model_id)
17
+
18
 
19
  def transcribe(audio_file_mic=None, audio_file_upload=None, language="Luganda (lug)"):
20
  if audio_file_mic: