Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 import os
 from datasets import load_dataset
-
+import pandas as pd
+import matplotlib.pyplot as plt
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
@@ -11,5 +12,21 @@ df = ds.to_pandas()
 
 st.title("MMLU Translations Progress")
 
+# Extract the language from the metadata column and create a new column
+df['language'] = df['metadata'].apply(lambda x: x.get('language'))
+
+# Count the occurrences of each language
+language_counts = df['language'].value_counts()
+
+# Plotting the bar chart using matplotlib
+fig, ax = plt.subplots()
+language_counts.plot(kind='bar', ax=ax)
+ax.set_title('Number of Rows for Each Language')
+ax.set_xlabel('Language')
+ax.set_ylabel('Count')
+
+# Display the plot in the Streamlit app
+st.pyplot(fig)
+
 st.dataframe(df)