change defaults in corpus.

corpus.py

@@ -5,7 +5,8 @@ from ferret import Benchmark
 from torch.nn.functional import softmax
 from copy import deepcopy
 
-DEFAULT_MODEL = "cardiffnlp/twitter-xlm-roberta-base-sentiment"
+DEFAULT_MODEL = "Hate-speech-CNERG/bert-base-uncased-hatexplain"
+DEFAULT_SAMPLES = "3,5,8,13,15,17,18,25,27,28"
 
 
 @st.cache()
@@ -58,12 +59,12 @@ def body():
 
     samples_string = st.text_input(
         "List of samples",
-        "11,6,42",
+        DEFAULT_SAMPLES,
         help="List of indices in the dataset, comma-separated.",
     )
     compute = st.button("Run")
 
-    samples = list(map(int, samples_string.split(",")))
+    samples = list(map(int, samples_string.replace(" ", "").split(",")))
 
     if compute and model_name:
 
@@ -77,8 +78,12 @@ def body():
             @st.cache(allow_output_mutation=True)
             def compute_table(samples):
                 data = bench.load_dataset("hatexplain")
-                sample_evaluations = bench.evaluate_samples(data, samples)
-                table = bench.show_samples_evaluation_table(sample_evaluations)
+                sample_evaluations = bench.evaluate_samples(
+                    data, samples, target=class_labels.index(target)
+                )
+                table = bench.show_samples_evaluation_table(sample_evaluations).format(
+                    "{:.2f}"
+                )
                 return table
 
             table = compute_table(samples)