Spaces:
Running
Running
Update app.py
Browse files
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 =
|
|
|
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:
|