Use lineplot instead of plot

app.py

@@ -7,8 +7,8 @@ import decord
 import nncore
 import torch
 import gradio as gr
-import matplotlib.pyplot as plt
 import numpy as np
+import pandas as pd
 import torchvision.transforms.functional as F
 from decord import VideoReader
 from nncore.engine import load_checkpoint
@@ -59,7 +59,7 @@ def init_model(config, checkpoint):
     return model, cfg
 
 
-def main(video, query, model, cfg, fig, ax):
+def main(video, query, model, cfg):
     if len(query) == 0:
         raise gr.Error('Text query can not be empty.')
 
@@ -81,33 +81,29 @@ def main(video, query, model, cfg, fig, ax):
 
     hd = pred['_out']['saliency'].cpu()
     hd = ((hd - hd.min()) / (hd.max() - hd.min())).tolist()
+    hd = pd.DataFrame(dict(x=range(0, len(hd) * 2, 2), y=hd))
 
-    ax.cla()
-    ax.plot(range(0, len(hd) * 2, 2), hd)
-
-    ax.set_xlabel('Time (s)', fontsize=15)
-    ax.set_ylabel('Saliency Score', fontsize=15)
-
-    ax.tick_params(labelsize=14)
-
-    return mr, fig
+    return mr, hd
 
 
 model, cfg = init_model(CONFIG, WEIGHT)
 
-plt.tight_layout(rect=(0.02, 0.05, 0.95, 0.885))
-fig, ax = plt.subplots(figsize=(10, 5.5))
-
 demo = gr.Interface(
-    fn=partial(main, model=model, cfg=cfg, fig=fig, ax=ax),
+    fn=partial(main, model=model, cfg=cfg),
     inputs=[gr.Video(label='Video'),
             gr.Textbox(label='Text Query')],
     outputs=[
         gr.Dataframe(
             headers=['Start Time', 'End Time', 'Score'], label='Moment Retrieval'),
-        gr.Plot(label='Highlight Detection')
+        gr.LinePlot(
+            x='x',
+            y='y',
+            x_title='Time (seconds)',
+            y_title='Saliency Score',
+            label='Highlight Detection')
     ],
     allow_flagging='never',
     title=TITLE,
     description=DESCRIPTION)
+
 demo.launch()

requirements.txt

@@ -1,6 +1,6 @@
 git+https://github.com/openai/CLIP.git@a1d0717
 decord==0.6.0
-matplotlib==3.9.0
 nncore==0.4.4
+pandas==2.2.2
 torch==2.2.1
 torchvision==0.17.1