feat: gradio fixes

app.py

@@ -3,7 +3,7 @@ from transformers import pipeline
 from transformers.pipelines.audio_utils import ffmpeg_read
 import gradio as gr
 
-MODEL_NAME = "oyemade/whisper-medium-yo"
+MODEL_NAME = "neoform-ai/whisper-medium-yoruba"
 BATCH_SIZE = 8
 
 device = 0 if torch.cuda.is_available() else "cpu"
@@ -55,16 +55,14 @@ demo = gr.Blocks()
 mic_transcribe = gr.Interface(
     fn=transcribe,
     inputs=[
-        gr.inputs.Audio(source="microphone", type="filepath", optional=True),
-        gr.inputs.Radio(["transcribe", "translate"], label="Task", default="transcribe"),
-        gr.inputs.Checkbox(default=False, label="Return timestamps"),
+        gr.Audio(sources="microphone", type="filepath"),
+        gr.Radio(["transcribe", "translate"], label="Task"),
+        gr.Checkbox(label="Return timestamps"),
     ],
     outputs="text",
-    layout="horizontal",
-    theme="huggingface",
     title="NeoForm AI Demo: Transcribe Yoruba Audio",
     description=(
-        "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the"
+        "Transcribe long-form microphone or audio inputs in Yoruba language with the click of a button! Demo uses the"
         f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
         " of arbitrary length."
     ),
@@ -74,23 +72,17 @@ mic_transcribe = gr.Interface(
 file_transcribe = gr.Interface(
     fn=transcribe,
     inputs=[
-        gr.inputs.Audio(source="upload", optional=True, label="Audio file", type="filepath"),
-        gr.inputs.Radio(["transcribe", "translate"], label="Task", default="transcribe"),
-        gr.inputs.Checkbox(default=False, label="Return timestamps"),
+        gr.Audio(sources="upload", label="Audio file", type="filepath"),
+        gr.Radio(["transcribe", "translate"], label="Task"),
+        gr.Checkbox(label="Return timestamps"),
     ],
     outputs="text",
-    layout="horizontal",
-    theme="huggingface",
     title="NeoForm AI Demo: Transcribe Yoruba Audio",
     description=(
-        "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the"
+        "Transcribe long-form microphone or audio inputs in Yoruba language with the click of a button! Demo uses the"
         f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
         " of arbitrary length."
     ),
-    examples=[
-        ["./example.flac", "transcribe", False],
-        ["./example.flac", "transcribe", True],
-    ],
     cache_examples=True,
     allow_flagging="never",
 )
@@ -98,4 +90,4 @@ file_transcribe = gr.Interface(
 with demo:
     gr.TabbedInterface([mic_transcribe, file_transcribe], ["Transcribe Microphone", "Transcribe Audio File"])
 
-demo.launch(enable_queue=True)
+demo.launch()