Spaces:
Runtime error
Runtime error
Commit
·
52f0ba0
1
Parent(s):
1e78e73
Update app.py
Browse files
app.py
CHANGED
@@ -1,13 +1,13 @@
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
import librosa
|
|
|
4 |
from transformers import pipeline
|
5 |
from stitched_model import CombinedModel
|
6 |
|
7 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
8 |
|
9 |
-
|
10 |
-
model = CombinedModel("facebook/mms-1b-all", "Sunbird/sunbird-mul-en-mbart-merged", device=device)
|
11 |
|
12 |
def transcribe(audio_file_mic=None, audio_file_upload=None):
|
13 |
if audio_file_mic:
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
import librosa
|
4 |
+
import json
|
5 |
from transformers import pipeline
|
6 |
from stitched_model import CombinedModel
|
7 |
|
8 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
9 |
|
10 |
+
model = CombinedModel("indonesian-nlp/wav2vec2-luganda", "Sunbird/sunbird-mul-en-mbart-merged", device=device)
|
|
|
11 |
|
12 |
def transcribe(audio_file_mic=None, audio_file_upload=None):
|
13 |
if audio_file_mic:
|