app.py

import streamlit as st
from datasets import load_dataset

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

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.text("Question")
    st.markdown(dataset["train"]["prompt"][len_index])
    st.text("Chosen")
    st.markdown(dataset["train"]["chosen"][len_index])
    st.text("Rejected")
    st.markdown(dataset["train"]["rejected"][len_index])