Update app.py

app.py

@@ -9,7 +9,7 @@ from pathlib import Path
 import cv2
 import pandas as pd
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
-import matplotlib.pyplot as plt
+
 
 #torch.hub.download_url_to_file('https://github.com/AaronCWacker/Yggdrasil/blob/main/images/BeautyIsTruthTruthisBeauty.JPG', 'BeautyIsTruthTruthisBeauty.JPG')
 #torch.hub.download_url_to_file('https://github.com/AaronCWacker/Yggdrasil/blob/main/images/PleaseRepeatLouder.jpg', 'PleaseRepeatLouder.jpg')
@@ -22,17 +22,6 @@ torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/e
 torch.hub.download_url_to_file('https://i.imgur.com/mwQFd7G.jpeg', 'Hindi.jpeg')
 
 
-def plot_temporal_profile(temporal_profile):
-    fig = plt.figure()
-    for i, profile in enumerate(temporal_profile):
-        x, y = zip(*profile)
-        plt.plot(x, y, label=f"Box {i+1}")
-    plt.title("Temporal Profiles")
-    plt.xlabel("Time (s)")
-    plt.ylabel("Value")
-    plt.legend()
-    return fig
-
 def draw_boxes(image, bounds, color='yellow', width=2):
     draw = ImageDraw.Draw(image)
     for bound in bounds:
@@ -187,10 +176,7 @@ def inference(video, lang, time_step, full_scan, number_filter, use_trocr, perio
             df_list.append({"Box": f"Box {i+1}", "Time (s)": t, "Text": text})
         df_list.append({"Box": f"", "Time (s)": "", "Text": ""})
     df = pd.concat([pd.DataFrame(df_list)])
-
-    # generate the plot of temporal profile
-    plot_fig = plot_temporal_profile(temporal_profiles)
-    return output, im, plot_fig, df
+    return output, im, df
 
 
 title = '🖼️Video to Multilingual OCR👁️Gradio'
@@ -228,8 +214,7 @@ gr.Interface(
     [
         gr.outputs.Video(label='Output Video'),
         gr.outputs.Image(label='Output Preview', type='numpy'),
-        gr.Plot(label='Temporal Profile'),
-        gr.outputs.Dataframe(headers=['Box', 'Time (s)', 'Text'], type='pandas')
+        gr.outputs.Dataframe(headers=['Box', 'Time (s)', 'Text'], type='pandas'),
     ],
     title=title,
     description=description,
@@ -237,4 +222,4 @@ gr.Interface(
     examples=examples,
     css=css,
     enable_queue=True
-).launch(debug=True, share=True)
+).launch(debug=True)