Add sv-en translation model

app.py

@@ -1,16 +1,19 @@
 from transformers import pipeline
 import gradio as gr
 
-pipe = pipeline(model="Scalable-ML/whisper-small-sv-dropout-6mb-checkpoint-2000")  
+transcription_pipe = pipeline(model="Scalable-ML/whisper-small-sv-dropout-6mb-checkpoint-2000")  
+translation_pipe = pipeline(model="Helsinki-NLP/opus-mt-sv-en")
 
-def transcribe(audio):
-    text = pipe(audio)["text"]
-    return text
+
+def transcribe_and_translate(audio):
+    transcription = transcription_pipe(audio)["text"]
+    translation = translation_pipe(transcription)["translation_text"]
+    return transcription, translation
 
 iface = gr.Interface(
     fn=transcribe,
     inputs=gr.Audio(source="microphone", type="filepath"),
-    outputs="text",
+    outputs=[gr.Textbox(), gr.Textbox()],
     title="Whisper Small Swedish to English Translator",
     description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model, and Swedish to English translation using a small T5 model",
 )