fiddle with plots
Browse files
app.py
CHANGED
@@ -138,6 +138,7 @@ BENCHMARK_DATA = {
|
|
138 |
},
|
139 |
},
|
140 |
}
|
|
|
141 |
|
142 |
|
143 |
def get_plot(model_name, plot_eager, generate_type):
|
@@ -147,7 +148,6 @@ def get_plot(model_name, plot_eager, generate_type):
|
|
147 |
if plot_eager == "No":
|
148 |
df = df[df["framework"] != "TF (Eager Execition)"]
|
149 |
|
150 |
-
plt.figure(dpi=1200)
|
151 |
g = sns.catplot(
|
152 |
data=df,
|
153 |
kind="bar",
|
@@ -160,7 +160,8 @@ def get_plot(model_name, plot_eager, generate_type):
|
|
160 |
g.despine(left=True)
|
161 |
g.set_axis_labels("GPU", "Generation time (ms)")
|
162 |
g.legend.set_title("Framework")
|
163 |
-
|
|
|
164 |
|
165 |
demo = gr.Blocks()
|
166 |
|
@@ -196,7 +197,7 @@ with demo:
|
|
196 |
interactive=True
|
197 |
)
|
198 |
plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
|
199 |
-
plot = gr.
|
200 |
model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
201 |
eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
202 |
with gr.TabItem("Sample"):
|
@@ -223,7 +224,7 @@ with demo:
|
|
223 |
interactive=True
|
224 |
)
|
225 |
plot_fn = functools.partial(get_plot, generate_type="Sample")
|
226 |
-
plot = gr.
|
227 |
model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
228 |
eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
229 |
with gr.TabItem("Beam Search"):
|
@@ -249,7 +250,7 @@ with demo:
|
|
249 |
interactive=True
|
250 |
)
|
251 |
plot_fn = functools.partial(get_plot, generate_type="Beam Search")
|
252 |
-
plot = gr.
|
253 |
model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
254 |
eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
255 |
with gr.TabItem("Benchmark Information"):
|
|
|
138 |
},
|
139 |
},
|
140 |
}
|
141 |
+
FIGURE_PATH = "plt.png"
|
142 |
|
143 |
|
144 |
def get_plot(model_name, plot_eager, generate_type):
|
|
|
148 |
if plot_eager == "No":
|
149 |
df = df[df["framework"] != "TF (Eager Execition)"]
|
150 |
|
|
|
151 |
g = sns.catplot(
|
152 |
data=df,
|
153 |
kind="bar",
|
|
|
160 |
g.despine(left=True)
|
161 |
g.set_axis_labels("GPU", "Generation time (ms)")
|
162 |
g.legend.set_title("Framework")
|
163 |
+
plt.savefig(FIGURE_PATH, dpi=1200)
|
164 |
+
return FIGURE_PATH
|
165 |
|
166 |
demo = gr.Blocks()
|
167 |
|
|
|
197 |
interactive=True
|
198 |
)
|
199 |
plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
|
200 |
+
plot = gr.Image(value=plot_fn("T5 Small", "Yes")) # Show plot when the gradio app is initialized
|
201 |
model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
202 |
eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
203 |
with gr.TabItem("Sample"):
|
|
|
224 |
interactive=True
|
225 |
)
|
226 |
plot_fn = functools.partial(get_plot, generate_type="Sample")
|
227 |
+
plot = gr.Image(value=plot_fn("T5 Small", "Yes")) # Show plot when the gradio app is initialized
|
228 |
model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
229 |
eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
230 |
with gr.TabItem("Beam Search"):
|
|
|
250 |
interactive=True
|
251 |
)
|
252 |
plot_fn = functools.partial(get_plot, generate_type="Beam Search")
|
253 |
+
plot = gr.Image(value=plot_fn("T5 Small", "Yes")) # Show plot when the gradio app is initialized
|
254 |
model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
255 |
eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
256 |
with gr.TabItem("Benchmark Information"):
|