Update app.py

app.py

@@ -21,7 +21,7 @@ samples = load_data(source)
 n_samples = len(samples)
 
 index_example = st.number_input(f"Index of the sample (out of {n_samples}):",  min_value=0, max_value=n_samples-1, value=0, step=1)
-st.markdown(f"Displaying source: {source}")
+st.markdown(f"Example belongs to source: {samples[index_example]['source']}")
 st.subheader("Prompt")
 st.markdown(samples[index_example]["prompt"])