updated interface to include non fine-tuned model

app.py

@@ -1,17 +1,22 @@
 from transformers import pipeline
 import gradio as gr
 
-pipe = pipeline(model="openai/whisper-small")  # change to "your-username/the-name-you-picked"
+pipe_fine = pipeline(model="zeihers-mart/whisper-small-swedish-basic") 
+pipe_raw = pipeline(model="openai/whisper-small") 
 
 def transcribe(audio):
-    text = pipe(audio)["text"]
-    return text
+    text_sv = pipe_fine(audio)["text"]
+    print(f"Audio transcribed: {text_sv}")
+    text_raw= pipe_raw(audio)["text"]
+    print(f"Text translated: {text_raw}")
+    return text_sv, text_raw
 
 iface = gr.Interface(
     fn=transcribe, 
-    inputs=gr.Audio(sources=["microphone"], type="filepath"), 
-    outputs="text",
-    title="Whisper Small Swedish",
+    inputs=gr.Audio(source="microphone", type="filepath"), 
+    outputs=[gr.Textbox(label="Fine-tuned transcription"),
+             gr.Textbox(label="Whisper Transcription")],
+    title="Finetuned Whisper Swedish Small",
     description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model.",
 )