loubnabnl HF staff commited on
Commit
7237bc9
1 Parent(s): 2028475

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -14,7 +14,9 @@ def plot_throughput(bs=1):
14
  df.loc[df['Models'].str.contains('StarCoder|SantaCoder'), 'color'] = 'orange'
15
  df.loc[df['Models'].str.contains('CodeGen'), 'color'] = 'pink'
16
  df.loc[df['Models'].str.contains('Replit'), 'color'] = 'purple'
17
-
 
 
18
  fig = go.Figure()
19
 
20
  for i in df.index:
 
14
  df.loc[df['Models'].str.contains('StarCoder|SantaCoder'), 'color'] = 'orange'
15
  df.loc[df['Models'].str.contains('CodeGen'), 'color'] = 'pink'
16
  df.loc[df['Models'].str.contains('Replit'), 'color'] = 'purple'
17
+ df.loc[df['Models'].str.contains('Wizard'), 'color'] = '#00b3b3'
18
+ df.loc[df['Models'].str.contains('CodeGeeX'), 'color'] = '#00cc00'
19
+
20
  fig = go.Figure()
21
 
22
  for i in df.index: