explorall commited on
Commit
219cefd
1 Parent(s): 3a24165

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ from transformers import pipeline
2
  import gradio as gr
3
  from pytube import YouTube
4
 
5
- transcription_pipe = pipeline(model="explorall/whisper-small-sv-dropout-6mb")
 
6
  translation_pipe = pipeline(model="Helsinki-NLP/opus-mt-sv-en")
7
 
8
  def transcribe_and_translate(audio):
 
2
  import gradio as gr
3
  from pytube import YouTube
4
 
5
+ #transcription_pipe = pipeline(model="explorall/whisper-small-sv-dropout-6mb")
6
+ transcription_pipe = pipeline(model="explorall/whisper-small-sv")
7
  translation_pipe = pipeline(model="Helsinki-NLP/opus-mt-sv-en")
8
 
9
  def transcribe_and_translate(audio):