remove plot

app.py

@@ -2,7 +2,6 @@ from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
 from transformers.utils import is_flash_attn_2_available
 import torch
 import gradio as gr
-import matplotlib.pyplot as plt
 import time
 import os
 
@@ -64,6 +63,7 @@ def transcribe(inputs):
         start_time = time.time()
         result = distil_pipe_forward(*args, **kwargs)
         distil_runtime = time.time() - start_time
+        distil_runtime = round(distil_runtime, 2)
         return result
 
     distil_pipe._forward = _forward_distil_time
@@ -75,34 +75,13 @@ def transcribe(inputs):
         start_time = time.time()
         result = pipe_forward(*args, **kwargs)
         runtime = time.time() - start_time
+        runtime = round(runtime, 2)
         return result
 
     pipe._forward = _forward_time
     text = pipe(inputs, batch_size=BATCH_SIZE)["text"]
 
-    # Create figure and axis
-    fig, ax = plt.subplots(figsize=(5, 5))
-
-    # Define bar width and positions
-    bar_width = 0.1
-    positions = [0, 0.1]  # Adjusted positions to bring bars closer
-
-    # Plot data
-    ax.bar(positions[0], distil_runtime, bar_width, edgecolor='black')
-    ax.bar(positions[1], runtime, bar_width, edgecolor='black')
-
-    # Set title, labels, and xticks
-    ax.set_ylabel('Transcription time (s)')
-    ax.set_xticks(positions)
-    ax.set_xticklabels(['Distil-Whisper', 'Whisper'])
-
-    # Gridlines and other styling
-    ax.grid(which='major', axis='y', linestyle='--', linewidth=0.5)
-
-    # Use tight layout to avoid overlaps
-    plt.tight_layout()
-
-    yield distil_text, distil_runtime, text, runtime, plt
+    yield distil_text, distil_runtime, text, runtime
 
 if __name__ == "__main__":
     with gr.Blocks() as demo:
@@ -129,18 +108,15 @@ if __name__ == "__main__":
         )
         audio = gr.components.Audio(type="filepath", label="Audio input")
         button = gr.Button("Transcribe")
-        plot = gr.components.Plot()
         with gr.Row():
             distil_runtime = gr.components.Textbox(label="Distil-Whisper Transcription Time (s)")
             runtime = gr.components.Textbox(label="Whisper Transcription Time (s)")
         with gr.Row():
             distil_transcription = gr.components.Textbox(label="Distil-Whisper Transcription", show_copy_button=True)
             transcription = gr.components.Textbox(label="Whisper Transcription", show_copy_button=True)
-
         button.click(
             fn=transcribe,
             inputs=audio,
-            outputs=[distil_transcription, distil_runtime, transcription, runtime, plot],
+            outputs=[distil_transcription, distil_runtime, transcription, runtime],
         )
-
-    demo.queue().launch()
+    demo.queue().launch()