fiddle with plots

app.py

@@ -138,6 +138,7 @@ BENCHMARK_DATA = {
         },
     },
 }
+FIGURE_PATH = "plt.png"
 
 
 def get_plot(model_name, plot_eager, generate_type):
@@ -147,7 +148,6 @@ def get_plot(model_name, plot_eager, generate_type):
     if plot_eager == "No":
         df = df[df["framework"] != "TF (Eager Execition)"]
 
-    plt.figure(dpi=1200)
     g = sns.catplot(
         data=df,
         kind="bar",
@@ -160,7 +160,8 @@ def get_plot(model_name, plot_eager, generate_type):
     g.despine(left=True)
     g.set_axis_labels("GPU", "Generation time (ms)")
     g.legend.set_title("Framework")
-    return plt.gcf()
+    plt.savefig(FIGURE_PATH, dpi=1200)
+    return FIGURE_PATH
 
 demo = gr.Blocks()
 
@@ -196,7 +197,7 @@ with demo:
                     interactive=True
                 )
             plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
-            plot = gr.Plot(value=plot_fn("T5 Small", "Yes"))  # Show plot when the gradio app is initialized
+            plot = gr.Image(value=plot_fn("T5 Small", "Yes"))  # Show plot when the gradio app is initialized
             model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
             eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
         with gr.TabItem("Sample"):
@@ -223,7 +224,7 @@ with demo:
                     interactive=True
                 )
             plot_fn = functools.partial(get_plot, generate_type="Sample")
-            plot = gr.Plot(value=plot_fn("T5 Small", "Yes"))  # Show plot when the gradio app is initialized
+            plot = gr.Image(value=plot_fn("T5 Small", "Yes"))  # Show plot when the gradio app is initialized
             model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
             eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
         with gr.TabItem("Beam Search"):
@@ -249,7 +250,7 @@ with demo:
                     interactive=True
                 )
             plot_fn = functools.partial(get_plot, generate_type="Beam Search")
-            plot = gr.Plot(value=plot_fn("T5 Small", "Yes"))  # Show plot when the gradio app is initialized
+            plot = gr.Image(value=plot_fn("T5 Small", "Yes"))  # Show plot when the gradio app is initialized
             model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
             eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
         with gr.TabItem("Benchmark Information"):