Update app.py

app.py

@@ -11,7 +11,7 @@ from datetime import datetime
 
 import assets
 
-def sendToWhisper(audio_record, audio_upload, task, models_selected, language_toggle, language_selected, without_timestamps):
+def sendToWhisper(audio_record, audio_upload, without_timestamps):
     results = []
 
     audio = None
@@ -24,25 +24,14 @@ def sendToWhisper(audio_record, audio_upload, task, models_selected, language_to
 
     audio = whisper.load_audio(audio)
     audio = whisper.pad_or_trim(audio)
-
-    for model_name in models_selected:
-        start = datetime.now()
-        model = whisper.load_model(model_name)
-        mel = whisper.log_mel_spectrogram(audio).to(model.device)
-        options = whisper.DecodingOptions(fp16 = False, without_timestamps=without_timestamps, task=task)
-        if language_toggle:
-            options = whisper.DecodingOptions(fp16 = False, without_timestamps=without_timestamps, task=task, language=language_selected)
-        language = ""
-        prob = 0
-        if model_name in assets.lang_detect:
-            _, probs = model.detect_language(mel)
-            language = max(probs, key=probs.get)
-            prob = probs[language]
-        else:
-            language="en"
-            options = whisper.DecodingOptions(fp16 = False, without_timestamps=without_timestamps, task=task, language="en")
-        output_text = whisper.decode(model, mel, options)
-        results.append([model_name, output_text.text, language, str(prob), str((datetime.now() - start).total_seconds())])
+    
+    start = datetime.now()
+    model = whisper.load_model(model_name)
+    mel = whisper.log_mel_spectrogram(audio).to(model.device)
+    # options = whisper.DecodingOptions(fp16 = False, without_timestamps=without_timestamps, task=task)
+    options = whisper.DecodingOptions()
+    output_text = whisper.decode(model, mel, options)
+    results.append([model_name, output_text.text, language, str(prob), str((datetime.now() - start).total_seconds())])
     return results
 
 avail_models = whisper.available_models()
@@ -77,6 +66,7 @@ with gr.Blocks(css=assets.css) as demo:
         gr.Markdown("## Output")
         output = gr.Dataframe(headers=["Model", "Text", "Language", "Language Confidence","Time(s)"], label="Results", wrap=True)
 
-    submit.click(fn=sendToWhisper, inputs=[audio_record, audio_upload, task, models_selected, language_toggle, language_selected, without_timestamps], outputs=output)
+    submit.click(fn=sendToWhisper, inputs=[audio_record, audio_upload, without_timestamps], outputs=output) 
+# [audio_record, audio_upload, task, models_selected, language_toggle, language_selected, without_timestamps], outputs=output)
 
 demo.launch()