joaogante HF staff commited on
Commit
f423b53
1 Parent(s): 34aac7c

fiddle with plots

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -147,6 +147,7 @@ def get_plot(model_name, plot_eager, generate_type):
147
  if plot_eager == "No":
148
  df = df[df["framework"] != "TF (Eager Execition)"]
149
 
 
150
  g = sns.catplot(
151
  data=df,
152
  kind="bar",
@@ -154,7 +155,7 @@ def get_plot(model_name, plot_eager, generate_type):
154
  y="value",
155
  hue="framework",
156
  palette={"PyTorch": "blue", "TF (Eager Execition)": "orange", "TF (XLA)": "red"},
157
- # height=6
158
  )
159
  g.despine(left=True)
160
  g.set_axis_labels("GPU", "Generation time (ms)")
 
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",
 
155
  y="value",
156
  hue="framework",
157
  palette={"PyTorch": "blue", "TF (Eager Execition)": "orange", "TF (XLA)": "red"},
158
+ alpha=.9,
159
  )
160
  g.despine(left=True)
161
  g.set_axis_labels("GPU", "Generation time (ms)")