Update app.py

app.py

@@ -1,18 +1,24 @@
 from transformers import pipeline
 import gradio as gr
+from song_guesser import SongGuesser
 
-pipe = pipeline(model="MarieGotthardt/whisper_tuned")  # change to "your-username/the-name-you-picked"
+pipe = pipeline(model="SamuelHarner/whisper-tuned")  # change to "your-username/the-name-you-picked"
 
 def transcribe(audio):
     text = pipe(audio)["text"]
     return text
 
+def get_song_guess(audio):
+    user_query = transcribe(audio)
+    output = "Vad vi hörde från dig: " + user_query + "\n\nVi tror att du sjöng: " + SongGuesser.guess_song(user_query)
+    return output
+
 iface = gr.Interface(
-    fn=transcribe,
+    fn=get_song_guess, 
     inputs=gr.Audio(sources=["microphone"], type="filepath"),
     outputs="text",
-    title="Whisper Small Swedish",
-    description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model.",
+    title="Sjung en sång och låt oss gissa  🎤 🎄",
+    description="Sing a Swedish Christmas song and see if we can guess it. If our guess is wrong, try singing more clearly, more of the lyrics, or another song and maybe we will get it the next time!",
 )
 
 iface.launch()