Update app.py

app.py

@@ -47,12 +47,12 @@ def get_prediction(inputs, **kwargs):
 
 def transcribe(inputs: str,
                add_punctuation: bool,
+               add_silence_end: bool,
+               add_silence_start: bool,
                num_speakers: float,
                min_speakers: float,
                max_speakers: float,
-               chunk_length_s: float,
-               add_silence_end: float,
-               add_silence_start: float):
+               chunk_length_s: float):
     if inputs is None:
         raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
     with open(inputs, "rb") as f:
@@ -65,8 +65,8 @@ def transcribe(inputs: str,
         min_speakers=int(min_speakers) if min_speakers != 0 else None,
         max_speakers=int(max_speakers) if max_speakers != 0 else None,
         chunk_length_s=int(chunk_length_s) if chunk_length_s != 30 else None,
-        add_silence_end=add_silence_end if add_silence_end != 0 else None,
-        add_silence_start=add_silence_start if add_silence_start != 0 else None
+        add_silence_end=0.5 if add_silence_end else None,
+        add_silence_start=0.5 if add_silence_start else None
     )
     output = ""
     for n, s in enumerate(prediction["speaker_ids"]):
@@ -79,8 +79,8 @@ description = (f"Transcribe and diarize long-form microphone or audio inputs wit
                f"Kotoba-Whisper [{model_name}](https://huggingface.co/{model_name}).")
 title = f"Audio Transcription and Diarization with {os.path.basename(model_name)}"
 shared_config = {"fn": transcribe, "title": title, "description": description, "allow_flagging": "never", "examples": [
-    [example_file, True, 0, 0, 0, 30, 0.5, 0.5],
-    [example_file, True, 4, 0, 0, 30, 0.5, 0.5]
+    [example_file, True, True, True 0, 0, 0, 30],
+    [example_file, True, True, True, 4, 0, 0, 30]
 ]}
 o_upload = gr.Markdown()
 o_mic = gr.Markdown()
@@ -91,12 +91,12 @@ i_upload = gr.Interface(
     inputs=[
         gr.Audio(sources="upload", type="filepath", label="Audio file"),
         gr.Checkbox(label="add punctuation", value=True),
+        gr.Checkbox(label="add silence at the end", value=True),
+        gr.Checkbox(label="add silence at the start", value=True),
         gr.Slider(0, 10, label="num speakers (set 0 for auto-detect mode)", value=0, step=1),
         gr.Slider(0, 10, label="min speakers (set 0 for auto-detect mode)", value=0, step=1),
         gr.Slider(0, 10, label="max speakers (set 0 for auto-detect mode)", value=0, step=1),
         gr.Slider(5, 30, label="chunk length for ASR", value=30, step=1),
-        gr.Slider(0, 0.5, label="silence at the end", value=0.5, step=0.05),
-        gr.Slider(0, 0.5, label="silence at the start", value=0.5, step=0.05),
     ],
     outputs=gr.Markdown(),
     **shared_config
@@ -105,12 +105,12 @@ i_mic = gr.Interface(
     inputs=[
         gr.Audio(sources="microphone", type="filepath", label="Microphone input"),
         gr.Checkbox(label="add punctuation", value=True),
+        gr.Checkbox(label="add silence at the end", value=True),
+        gr.Checkbox(label="add silence at the start", value=True),
         gr.Slider(0, 10, label="num speakers (set 0 for auto-detect mode)", value=0, step=1),
         gr.Slider(0, 10, label="min speakers (set 0 for auto-detect mode)", value=0, step=1),
         gr.Slider(0, 10, label="max speakers (set 0 for auto-detect mode)", value=0, step=1),
         gr.Slider(5, 30, label="chunk length for ASR", value=30, step=1),
-        gr.Slider(0, 0.5, label="silence at the end", value=0.5, step=0.05),
-        gr.Slider(0, 0.5, label="silence at the start", value=0.5, step=0.05),
     ],
     outputs=gr.Markdown(),
     **shared_config