Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -79,7 +79,6 @@ def main():
|
|
79 |
# Task selection
|
80 |
st.sidebar.write("#### 1. Select task(s) to view models")
|
81 |
selected_tasks = st.sidebar.multiselect("", options=task_order, default=task_order)
|
82 |
-
|
83 |
# Mapping from task to CSV file name.
|
84 |
task_to_file = {
|
85 |
"Text Generation": "text_gen_energyscore.csv",
|
@@ -93,10 +92,7 @@ def main():
|
|
93 |
"Question Answering": "question_answering_energyscore.csv",
|
94 |
"Sentence Similarity": "sentence_similarity_energyscore.csv"
|
95 |
}
|
96 |
-
|
97 |
st.sidebar.write("#### 2. Select a model to generate label")
|
98 |
-
|
99 |
-
|
100 |
default_model_data = {
|
101 |
'provider': "AI Provider",
|
102 |
'model': "Model Name",
|
@@ -185,7 +181,6 @@ def main():
|
|
185 |
st.sidebar.write("- [FAQ](https://huggingface.github.io/AIEnergyScore/#faq)")
|
186 |
st.sidebar.write("- [Documentation](https://huggingface.github.io/AIEnergyScore/#documentation)")
|
187 |
|
188 |
-
|
189 |
def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
|
190 |
bg_resized = background_image.resize(final_size, Image.Resampling.LANCZOS)
|
191 |
draw = ImageDraw.Draw(bg_resized)
|
@@ -200,7 +195,7 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
|
|
200 |
|
201 |
title_x, title_y = 33, 150
|
202 |
details_x, details_y = 480, 256
|
203 |
-
energy_x, energy_y =
|
204 |
|
205 |
draw.text((title_x, title_y), str(model_data['provider']), font=title_font, fill="black")
|
206 |
draw.text((title_x, title_y + 38), str(model_data['model']), font=title_font, fill="black")
|
|
|
79 |
# Task selection
|
80 |
st.sidebar.write("#### 1. Select task(s) to view models")
|
81 |
selected_tasks = st.sidebar.multiselect("", options=task_order, default=task_order)
|
|
|
82 |
# Mapping from task to CSV file name.
|
83 |
task_to_file = {
|
84 |
"Text Generation": "text_gen_energyscore.csv",
|
|
|
92 |
"Question Answering": "question_answering_energyscore.csv",
|
93 |
"Sentence Similarity": "sentence_similarity_energyscore.csv"
|
94 |
}
|
|
|
95 |
st.sidebar.write("#### 2. Select a model to generate label")
|
|
|
|
|
96 |
default_model_data = {
|
97 |
'provider': "AI Provider",
|
98 |
'model': "Model Name",
|
|
|
181 |
st.sidebar.write("- [FAQ](https://huggingface.github.io/AIEnergyScore/#faq)")
|
182 |
st.sidebar.write("- [Documentation](https://huggingface.github.io/AIEnergyScore/#documentation)")
|
183 |
|
|
|
184 |
def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
|
185 |
bg_resized = background_image.resize(final_size, Image.Resampling.LANCZOS)
|
186 |
draw = ImageDraw.Draw(bg_resized)
|
|
|
195 |
|
196 |
title_x, title_y = 33, 150
|
197 |
details_x, details_y = 480, 256
|
198 |
+
energy_x, energy_y = 430, 472
|
199 |
|
200 |
draw.text((title_x, title_y), str(model_data['provider']), font=title_font, fill="black")
|
201 |
draw.text((title_x, title_y + 38), str(model_data['model']), font=title_font, fill="black")
|