update

app.py

@@ -4,7 +4,7 @@ import glob
 import os
 from datetime import datetime
 
-st.set_page_config(layout="wide")
+#st.set_page_config(layout="wide")
 st.title('Meta Open LLM leaderboard')
 
 data_path = './data/20230820_0354/'
@@ -16,10 +16,39 @@ parsed_date = datetime.strptime(date_str, "%Y%m%d_%H%M")
 # Formatting the parsed date
 formatted_date = parsed_date.strftime("%b %d, %Y %H:%M")
 
-st.subheader("Generated on: " + formatted_date)
+captions_map = {
+    "hg_average_to_agentbench_compare.png": "HF to AgentBench compare",
+    "hg_average_to_opencompass_compare.png": "HF to OpenCompass compare",
+    "hg_average_to_mt_bench_compare.png": "HF to MT-Bench compare",
+    "hg_average_to_mosaic_compare.png": "HF to MosaicML compare",
+    "hg_average_to_alpacaeval_compare.png": "HF to AlpacaEval compare"
+}
+
+st.write("Generated on: " + formatted_date)
+st.divider()
+
 imgs = glob.glob(os.path.join(data_path, '*.png'))
 
+# Extracting images that start with "hf_llm_diagram"
+hf_llm_diagrams = [img for img in imgs if 'hf_llm_diagram' in os.path.basename(img)]
+
+# Getting the remaining images
+remaining_imgs = [img for img in imgs if 'hf_llm_diagram' not in os.path.basename(img)]
+
+st.write("HuggingFace Open LLM leaderboard by Model Size")
+st.image(hf_llm_diagrams,use_column_width="auto")
+
+st.divider()
+st.write("HuggingFace and Other Leaderboards: A Comparative Model Evaluation")
+st.caption("Only models evaluated on both leaderboards are included.")
 cols = st.columns(2)
 
-for i, img in enumerate(imgs):
-    cols[i % 2].image(img, width=None)
+for i, img in enumerate(remaining_imgs):
+    # Extract the filename from the full image path
+    filename = os.path.basename(img)
+
+    # Get the caption from the captions_map dictionary
+    caption = captions_map.get(filename, "")  # If no caption is found, it will default to an empty string
+
+    # Display the image with the caption
+    cols[i % 2].image(img, caption=caption, width=None)