Upload 3 files

README.md

@@ -1,8 +1,8 @@
 ---
-title: ID2223Lab2
-emoji: 📈
-colorFrom: gray
-colorTo: indigo
+title: ID2223 Lab2 Whisper
+emoji: 🐨
+colorFrom: purple
+colorTo: purple
 sdk: gradio
 sdk_version: 3.12.0
 app_file: app.py

app.py

@@ -0,0 +1,20 @@
+from transformers import pipeline
+import gradio as gr
+
+pipe1 = pipeline(model="khalidey/ID2223_Lab2_Whisper_SV")  # change to "your-username/the-name-you-picked"
+pipe2 = pipeline('text-generation', model='birgermoell/swedish-gpt')
+
+def transcribe(audio):
+    text = pipe1(audio)["text"]
+    generated_text = pipe2(text, max_length = 30, num_return_sequences=2)[0]['generated_text']
+    return text, generated_text
+
+iface = gr.Interface(
+    fn=transcribe, 
+    inputs=gr.Audio(source="microphone", type="filepath"), 
+    outputs=['text', 'text'],
+    title="Whisper Small Swedish + Swedish GPT",
+    description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model and text generation with Swedish GPT.",
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+transformers
+torch