Update app.py

app.py

@@ -205,8 +205,12 @@ 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  
 
-    draw.text((title_x, title_y), str(model_data['provider']), font=title_font, fill="black")
-    draw.text((title_x, title_y + 38), str(model_data['model']), font=title_font, fill="black")
+    # Capitalize the first letter of provider and model
+    provider_text = str(model_data['provider']).capitalize()
+    model_text = str(model_data['model']).capitalize()
+
+    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):
@@ -223,7 +227,5 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
 
     return bg_resized
 
-
-
 if __name__ == "__main__":
-    main()
+    main()