Update app.py

app.py

@@ -43,10 +43,10 @@ def process_with_fallback(func, *args, **kwargs):
             raise
 
 @spaces.GPU(duration=60)
-def transcribe_audio(audio_file, translate, model_size, use_diarization):
+def transcribe_audio(audio_file, translate, model_size, use_diarization=False):
     logger.info(f"Starting transcription: translate={translate}, model_size={model_size}, use_diarization={use_diarization}")
     try:
-        result = process_with_fallback(process_audio, audio_file, translate=translate, model_size=model_size, use_diarization=use_diarization)
+        result = process_with_fallback(process_audio, audio_file, translate=translate, model_size=model_size) # use_diarization=use_diarization
         logger.info("Transcription completed successfully")
         return result
     except Exception as e:
@@ -67,7 +67,7 @@ def summarize_text(text):
         return "Error occurred during summarization. Please try again."
 
 @spaces.GPU(duration=60)
-def process_and_summarize(audio_file, translate, model_size, use_diarization, do_summarize):
+def process_and_summarize(audio_file, translate, model_size, use_diarization=False, do_summarize=True):
     logger.info(f"Starting process_and_summarize: translate={translate}, model_size={model_size}, use_diarization={use_diarization}, do_summarize={do_summarize}")
     try:
         language_segments, final_segments = transcribe_audio(audio_file, translate, model_size, use_diarization)
@@ -104,7 +104,7 @@ with gr.Blocks() as iface:
     audio_input = gr.Audio(type="filepath")
     translate_checkbox = gr.Checkbox(label="Enable Translation")
     summarize_checkbox = gr.Checkbox(label="Enable Summarization", interactive=False)
-    diarization_checkbox = gr.Checkbox(label="Enable Speaker Diarization")
+    # diarization_checkbox = gr.Checkbox(label="Enable Speaker Diarization")
     model_dropdown = gr.Dropdown(choices=["tiny", "base", "small", "medium", "large", "large-v2", "large-v3"], label="Whisper Model Size", value="small")
     process_button = gr.Button("Process Audio")
     transcription_output = gr.Textbox(label="Transcription/Translation")