Update app.py

app.py

@@ -1,3 +1,41 @@
 import gradio as gr
 
-gr.Interface.load("models/burraco135/whisper-tiny-en").launch()
+from transformers import pipeline
+
+model_id = "burraco135/whisper-small-en"
+pipe = pipeline("automatic-speech-recognition", model=model_id)
+
+def transcribe_speech(filepath):
+ output = pipe(
+ filepath,
+ max_new_tokens=256,
+ generate_kwargs={
+ "task": "transcribe",
+ "language": "sinhalese",
+ }, # update with the language you've fine-tuned on
+ chunk_length_s=30,
+ batch_size=8,
+ )
+ return output["text"]
+
+demo = gr.Blocks()
+
+mic_transcribe = gr.Interface(
+ fn=transcribe_speech,
+ inputs=gr.Audio(source="microphone", type="filepath"),
+ outputs=gr.outputs.Textbox(),
+)
+
+file_transcribe = gr.Interface(
+ fn=transcribe_speech,
+ inputs=gr.Audio(source="upload", type="filepath"),
+ outputs=gr.outputs.Textbox(),
+)
+
+with demo:
+ gr.TabbedInterface(
+ [mic_transcribe, file_transcribe],
+ ["Transcribe Microphone", "Transcribe Audio File"],
+ )
+
+demo.launch(debug=True)