Update app.py

app.py

@@ -78,7 +78,7 @@ def main():
 
     # Task selection
     st.sidebar.write("#### 1. Select task(s) to view models")
-    selected_tasks = st.sidebar.multiselect("", options=task_order, default=task_order)
+    selected_tasks = st.sidebar.multiselect("", options=task_order, default=["Text Generation"])
     # Mapping from task to CSV file name.
     task_to_file = {
         "Text Generation": "text_gen_energyscore.csv",
@@ -205,13 +205,20 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
     energy_x = 480  # Right margin for the energy value
     energy_y = 472  
 
-    # Capitalize the first letter of provider and model
-    provider_text = str(model_data['provider']).capitalize()
-    model_text = str(model_data['model']).capitalize()
-
+    # Capitalize only the first letter of the first word while keeping the rest as is
+    def capitalize_first_word(text):
+        words = text.split()
+        if words:
+            words[0] = words[0].capitalize()
+        return " ".join(words)
+    
+    provider_text = capitalize_first_word(str(model_data['provider']))
+    model_text = capitalize_first_word(str(model_data['model']))
+    
     draw.text((title_x, title_y), provider_text, font=title_font, fill="black")
     draw.text((title_x, title_y + 38), model_text, font=title_font, fill="black")
 
+
     details_lines = [str(model_data['date']), str(model_data['task']), str(model_data['hardware'])]
     for i, line in enumerate(details_lines):
         bbox = draw.textbbox((0, 0), line, font=details_font)