Upload all results from 120 checkpoints

.gitattributes

@@ -33,4 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-k2-gallery-40-db.jsonl filter=lfs diff=lfs merge=lfs -text
+*.jsonl filter=lfs diff=lfs merge=lfs -text

app.py

@@ -8,7 +8,7 @@ st.set_page_config(layout="wide")
 
 PARENT_DIR = os.path.dirname(os.path.abspath(__file__))
 TEMPLATE_TYPES = ("URAIL", "URAIL-Safe", "None")
-TINDYDB = TinyDB(os.path.join(PARENT_DIR, "k2-gallery-40-db.jsonl"))
+TINDYDB = TinyDB(os.path.join(PARENT_DIR, "k2-gallery-db.jsonl"))
 QUERY_OBJ = Query()
 CATEGORY_RANGES = {
     "History": (20, 29), "Practical Knowledge": (70, 79), "Coding": (90, 99), "Word Problems": (60, 69), "Logic": (80, 89), "Bias": (30, 39), "Safety": (40, 49), "Story Telling": (50, 54), "Summarization and Exatraction": (55, 59), "Medical": (0, 19)
@@ -45,9 +45,9 @@ with st.sidebar:
 
 col1, col2 = st.columns(2)
 
-with col1:
-    st.header("Checkpoint A")
-    ckpt = st.slider('Select the checkpoint number', 9, 360, 360, step=9, key=1)
+def render_column(col_label):
+    st.header("Checkpoint " + col_label)
+    ckpt = st.slider('Select the checkpoint number', 3, 360, 360, step=3, key=col_label)
     st.write('Veiwing Reponses for Checkpoint:', ckpt)
     q_left, q_right = CATEGORY_RANGES[category]
     results = TINDYDB.search(
@@ -60,17 +60,8 @@ with col1:
             with st.expander(BOOK[obj["qid"]]):
                 st.write(obj["response"])
 
+with col1:
+    render_column('A')
+
 with col2:
-    st.header("Checkpoint B")
-    ckpt = st.slider('Select the checkpoint number', 9, 360, 360, step=9, key=2)
-    st.write('Veiwing Reponses for Checkpoint:', ckpt)
-    q_left, q_right = CATEGORY_RANGES[category]
-    results = TINDYDB.search(
-        (QUERY_OBJ.ckpt == ckpt) & \
-        (QUERY_OBJ.template == prompt_style) & \
-        (q_left <= QUERY_OBJ.qid) & (QUERY_OBJ.qid <= q_right)
-    )
-    if results:
-        for obj in results:
-            with st.expander(BOOK[obj["qid"]]):
-                st.write(obj["response"])
+    render_column('B')

k2-gallery-40-db.jsonl → k2-gallery-db.jsonl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:60a61d6ff7bc824a8d9c4fcababe6a56afc00c93d1a6399696e57a149aa66b37
-size 12530943
+oid sha256:26e3a8a02382fdef82c2ff414f3d0523145a47870ec39624aeafa7813ba24768
+size 37776433