sasha HF staff commited on
Commit
d4ded0a
β€’
1 Parent(s): f913e34

adding model names

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -10,7 +10,10 @@ def get_plots(task_data):
10
  task_df = px.scatter(task_df, x="model", y="total_gpu_energy (Wh)", height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
11
  return task_df
12
 
13
- # %% app.ipynb 3
 
 
 
14
  demo = gr.Blocks()
15
 
16
  with demo:
@@ -32,7 +35,11 @@ with demo:
32
  )
33
  with gr.TabItem("Text Classification 🎭"):
34
  with gr.Row():
35
- plot = gr.Plot(get_plots('data/text_classification.csv'))
 
 
 
 
36
  with gr.TabItem("Image Classification πŸ–ΌοΈ"):
37
  with gr.Row():
38
  landscape_data = gr.components.Dataframe(
 
10
  task_df = px.scatter(task_df, x="model", y="total_gpu_energy (Wh)", height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
11
  return task_df
12
 
13
+ def get_model_names(task_data):
14
+ task_df= pd.read_csv(task_data)
15
+ return task_df['model']
16
+
17
  demo = gr.Blocks()
18
 
19
  with demo:
 
35
  )
36
  with gr.TabItem("Text Classification 🎭"):
37
  with gr.Row():
38
+ with gr.Column():
39
+ plot = gr.Plot(get_plots('data/text_classification.csv'))
40
+ with gr.Column():
41
+ table = gr.Dataframe(get_model_names'('data/text_classification.csv'))
42
+
43
  with gr.TabItem("Image Classification πŸ–ΌοΈ"):
44
  with gr.Row():
45
  landscape_data = gr.components.Dataframe(