File size: 946 Bytes
08fba49
 
 
 
b8dbcb9
08fba49
 
 
 
 
46a52fc
55f3e96
46a52fc
73f5702
 
 
 
 
 
7ae7485
73f5702
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import streamlit as st
from datasets import load_dataset

dataset = load_dataset("reshinthadith/pairwise-code-review-instruct-critique-revision-python")
total_length = len(dataset["train"])

st.title('Pairwise Comparison')
with st.form("dataset_form"):
    len_index = st.slider('Dataset Index', 0, total_length,0)
    if st.form_submit_button("Load"):
        st.markdown("## Question")
        st.markdown(dataset["train"]["prompt"][len_index],unsafe_allow_html=True)
        st.markdown("## Chosen")
        chosen_text = dataset["train"]["chosen"][len_index]
        if "Critique" in chosen_text:
            st.text(chosen_text.split("Critique")[1])
        else:
            st.text(chosen_text)
        rejected_text = dataset["train"]["rejected"][len_index]
        st.markdown("## Rejected")
        if "Critique" in rejected_text:
            st.text(rejected_text.split("Critique")[1])
        else:
            st.text(rejected_text)