allandclive commited on
Commit
52f0ba0
·
1 Parent(s): 1e78e73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # Use facebook/mms-1b-all model for ASR and Sunbird/sunbird-mul-en-mbart-merged for translation
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: