Add subset filtering in first 10 rows

clarin_datasets/aspectemo_dataset.py

@@ -65,8 +65,9 @@ class AspectEmoDataset(DatasetToShow):
         labels_all = [x for subarray in labels_all for x in subarray]
 
         with dataframe_head:
-            df_to_show = full_dataframe.head(10)
-            st.header("First 10 observations of the dataset")
+            st.header("First 10 observations of the chosen subset")
+            selected_subset = st.selectbox(label="Select subset to see", options=self.subsets)
+            df_to_show = self.data_dict[selected_subset].head(10)
             st.dataframe(df_to_show)
             st.text_area(label="LaTeX code", value=df_to_show.style.to_latex())
 

clarin_datasets/kpwr_ner_datasets.py

@@ -78,8 +78,9 @@ class KpwrNerDataset(DatasetToShow):
         labels_all = [x for subarray in labels_all for x in subarray]
 
         with dataframe_head:
-            df_to_show = full_dataframe.head(10)
-            st.header("First 10 observations of the dataset")
+            st.header("First 10 observations of the chosen subset")
+            selected_subset = st.selectbox(label="Select subset to see", options=self.subsets)
+            df_to_show = self.data_dict[selected_subset].head(10)
             st.dataframe(df_to_show)
             st.text_area(label="LaTeX code", value=df_to_show.style.to_latex())