Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -78,7 +78,7 @@ def main():
|
|
78 |
|
79 |
# Task selection
|
80 |
st.sidebar.write("#### 1. Select task(s) to view models")
|
81 |
-
selected_tasks = st.sidebar.multiselect("", options=task_order, default=
|
82 |
# Mapping from task to CSV file name.
|
83 |
task_to_file = {
|
84 |
"Text Generation": "text_gen_energyscore.csv",
|
@@ -205,13 +205,20 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
|
|
205 |
energy_x = 480 # Right margin for the energy value
|
206 |
energy_y = 472
|
207 |
|
208 |
-
# Capitalize the first letter of
|
209 |
-
|
210 |
-
|
211 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
212 |
draw.text((title_x, title_y), provider_text, font=title_font, fill="black")
|
213 |
draw.text((title_x, title_y + 38), model_text, font=title_font, fill="black")
|
214 |
|
|
|
215 |
details_lines = [str(model_data['date']), str(model_data['task']), str(model_data['hardware'])]
|
216 |
for i, line in enumerate(details_lines):
|
217 |
bbox = draw.textbbox((0, 0), line, font=details_font)
|
|
|
78 |
|
79 |
# Task selection
|
80 |
st.sidebar.write("#### 1. Select task(s) to view models")
|
81 |
+
selected_tasks = st.sidebar.multiselect("", options=task_order, default=["Text Generation"])
|
82 |
# Mapping from task to CSV file name.
|
83 |
task_to_file = {
|
84 |
"Text Generation": "text_gen_energyscore.csv",
|
|
|
205 |
energy_x = 480 # Right margin for the energy value
|
206 |
energy_y = 472
|
207 |
|
208 |
+
# Capitalize only the first letter of the first word while keeping the rest as is
|
209 |
+
def capitalize_first_word(text):
|
210 |
+
words = text.split()
|
211 |
+
if words:
|
212 |
+
words[0] = words[0].capitalize()
|
213 |
+
return " ".join(words)
|
214 |
+
|
215 |
+
provider_text = capitalize_first_word(str(model_data['provider']))
|
216 |
+
model_text = capitalize_first_word(str(model_data['model']))
|
217 |
+
|
218 |
draw.text((title_x, title_y), provider_text, font=title_font, fill="black")
|
219 |
draw.text((title_x, title_y + 38), model_text, font=title_font, fill="black")
|
220 |
|
221 |
+
|
222 |
details_lines = [str(model_data['date']), str(model_data['task']), str(model_data['hardware'])]
|
223 |
for i, line in enumerate(details_lines):
|
224 |
bbox = draw.textbbox((0, 0), line, font=details_font)
|