Update app.py

app.py

@@ -11,6 +11,13 @@ with st.form("dataset_form"):
         st.markdown("## Question")
         st.text(dataset["train"]["prompt"][len_index])
         st.markdown("## Chosen")
-        st.text(dataset["train"]["chosen"][len_index].split("Critique")[1])
-        st.markdown("## Rejected")
-        st.text(dataset["train"]["rejected"][len_index].split("Critique")[1])
+        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]
+        if "Critique" in rejected_text:
+            st.text(rejected_text.split("Critique")[1])
+        else:
+            st.text(rejected_text)