Spaces:
Running
Running
wow, that was a rookie mistake
Browse files
app.py
CHANGED
|
@@ -3,7 +3,8 @@ import pandas as pd
|
|
| 3 |
from huggingface_hub import list_models
|
| 4 |
import plotly.express as px
|
| 5 |
|
| 6 |
-
def get_plots(
|
|
|
|
| 7 |
task_df['total_gpu_energy (Wh)'] = task_df['total_gpu_energy']*1000
|
| 8 |
task_df['energy_star'] = pd.cut(task_df['total_gpu_energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
|
| 9 |
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"})
|
|
|
|
| 3 |
from huggingface_hub import list_models
|
| 4 |
import plotly.express as px
|
| 5 |
|
| 6 |
+
def get_plots(task_data):
|
| 7 |
+
task_df= pd.read_csv(task_data)
|
| 8 |
task_df['total_gpu_energy (Wh)'] = task_df['total_gpu_energy']*1000
|
| 9 |
task_df['energy_star'] = pd.cut(task_df['total_gpu_energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
|
| 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"})
|