bgamazay commited on
Commit
dbb7425
·
verified ·
1 Parent(s): 18d0a50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
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=task_order)
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 provider and model
209
- provider_text = str(model_data['provider']).capitalize()
210
- model_text = str(model_data['model']).capitalize()
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)