up

app.py

@@ -102,6 +102,7 @@ if not os.path.isfile(DATA_FILE):
 with open(DATA_FILE, "r") as f:
     dataframe = pd.read_csv(DATA_FILE)
 
+dataframe[dataframe["modality"] == "audio"]
 int_downloads = np.array(
     [int(x.replace(",", "")) for x in dataframe["num_downloads"].values]
 )
@@ -143,8 +144,57 @@ bar_chart = (
 st.title("Bottom 20 downloads last 30 days")
 st.altair_chart(bar_chart, use_container_width=True)
 
-import ipdb; ipdb.set_trace()
+# print audio
+df_audio = dataframe[dataframe["modality"] == "audio"]
+audio_int_downloads = np.array(
+    [int(x.replace(",", "")) for x in df_audio["num_downloads"].values]
+)
+source = pd.DataFrame(
+    {
+        "Number of total downloads": audio_int_downloads,
+        "Model architecture name": df_audio["model_names"].values,
+    }
+)
+bar_chart = (
+    alt.Chart(source)
+    .mark_bar()
+    .encode(
+        y="Number of total downloads",
+        x=alt.X("Model architecture name", sort=None),
+    )
+)
+st.title("Audio downloads last 30 days")
+st.altair_chart(bar_chart, use_container_width=True)
+
+# print vision
+df_vision = dataframe[dataframe["modality"] == "vision"]
+vision_int_downloads = np.array(
+    [int(x.replace(",", "")) for x in df_vision["num_downloads"].values]
+)
+source = pd.DataFrame(
+    {
+        "Number of total downloads": vision_int_downloads,
+        "Model architecture name": df_vision["model_names"].values,
+    }
+)
+bar_chart = (
+    alt.Chart(source)
+    .mark_bar()
+    .encode(
+        y="Number of total downloads",
+        x=alt.X("Model architecture name", sort=None),
+    )
+)
+st.title("Vision downloads last 30 days")
+st.altair_chart(bar_chart, use_container_width=True)
+
 
 # print all stats
 st.title("All stats last 30 days")
 st.table(dataframe)
+
+st.title("All audio stats last 30 days")
+st.table(dataframe[dataframe["modality"] == "audio"])
+
+st.title("All vision stats last 30 days")
+st.table(dataframe[dataframe["modality"] == "vision"])