[email protected] commited on
Commit
8be0caa
·
1 Parent(s): a5e30a7
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -5,6 +5,8 @@ from apscheduler.schedulers.background import BackgroundScheduler
5
  from huggingface_hub import snapshot_download
6
  import plotly.graph_objects as go
7
  import plotly.express as px
 
 
8
  from src.about import (
9
  CITATION_BUTTON_LABEL,
10
  CITATION_BUTTON_TEXT,
@@ -72,8 +74,16 @@ ASSET_LEADERBOARD_DF = get_leaderboard_df(EVAL_RESULTS_PATH, EVAL_REQUESTS_PATH,
72
  def init_asset_plot(df):
73
  acc_col = 'Average ⬆️'
74
  df = df.sort_values(acc_col).tail(4)
75
- fig = px.line_polar(df, r='r', theta='theta', line_close=True)
76
- fig.update_traces(fill='toself')
 
 
 
 
 
 
 
 
77
  fig.update_layout(
78
  autosize=False,
79
  width=650,
 
5
  from huggingface_hub import snapshot_download
6
  import plotly.graph_objects as go
7
  import plotly.express as px
8
+ from src.about import Tasks, AssetTasks
9
+
10
  from src.about import (
11
  CITATION_BUTTON_LABEL,
12
  CITATION_BUTTON_TEXT,
 
74
  def init_asset_plot(df):
75
  acc_col = 'Average ⬆️'
76
  df = df.sort_values(acc_col).tail(4)
77
+ fig = go.Figure()
78
+ asset_tasks = [task.value.col_name for task in AssetTasks]
79
+ for _, row in df.iterrows():
80
+ fig.add_trace(go.Scatterpolar(
81
+ r=[row[asset_task] for asset_task in asset_tasks],
82
+ theta=asset_tasks,
83
+ fill='toself',
84
+ name=row['Model']
85
+ )
86
+ )
87
  fig.update_layout(
88
  autosize=False,
89
  width=650,