jtlonsako commited on
Commit
d29d0d4
1 Parent(s): a95e64c
Files changed (2) hide show
  1. app.py +0 -3
  2. requirements.txt +1 -0
app.py CHANGED
@@ -8,10 +8,7 @@ import gc
8
  import gradio as gr
9
  import librosa
10
  from transformers import Wav2Vec2ForCTC, Wav2Vec2ProcessorWithLM, AutoModelForSeq2SeqLM, AutoTokenizer
11
- from GPUtil import showUtilization as gpu_usage
12
  from numba import cuda
13
- from google.cloud import translate
14
-
15
 
16
  # load pretrained model
17
  model = Wav2Vec2ForCTC.from_pretrained("facebook/mms-1b-all")
 
8
  import gradio as gr
9
  import librosa
10
  from transformers import Wav2Vec2ForCTC, Wav2Vec2ProcessorWithLM, AutoModelForSeq2SeqLM, AutoTokenizer
 
11
  from numba import cuda
 
 
12
 
13
  # load pretrained model
14
  model = Wav2Vec2ForCTC.from_pretrained("facebook/mms-1b-all")
requirements.txt CHANGED
@@ -4,3 +4,4 @@ pyctcdecode
4
  torch
5
  librosa
6
  numba
 
 
4
  torch
5
  librosa
6
  numba
7
+ kenlm