Rachid Ammari commited on
Commit
bb28acc
1 Parent(s): 9a3ba32

removed forced GPU

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ from transformers import pipeline
2
  import gradio as gr
3
  import whisper
4
 
5
- wav2vec_en_model = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-base-960h", device=0)
6
- wav2vec_fr_model = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-large-xlsr-53-french", device=0)
7
  whisper_model = whisper.load_model("base")
8
 
9
  def transcribe_audio(language=None, mic=None, file=None):
 
2
  import gradio as gr
3
  import whisper
4
 
5
+ wav2vec_en_model = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-base-960h")
6
+ wav2vec_fr_model = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-large-xlsr-53-french")
7
  whisper_model = whisper.load_model("base")
8
 
9
  def transcribe_audio(language=None, mic=None, file=None):