Add OLM model as example. Fix func name typo

app.py

@@ -20,9 +20,6 @@ MODEL_NAME_DICT = {
 }
 MODEL_NAMES = list(MODEL_NAME_DICT.keys())
 
-# MODEL_NAMES = ["bert-base-uncased", "roberta-base", "bert-large-uncased", "roberta-large"]
-# OWN_MODEL_NAME = 'add-a-model'
-
 DECIMAL_PLACES = 1
 EPS = 1e-5  # to avoid /0 errors
 
@@ -153,7 +150,7 @@ models = {m : pipeline("fill-mask", model=m) for m in MODEL_NAMES if m != OWN_MO
 # %%
 
 
-def get_gendered_token_ids():
+def get_gendered_tokens():
     male_gendered_tokens = [list[0] for list in GENDERED_LIST]
     female_gendered_tokens = [list[1] for list in GENDERED_LIST]
 
@@ -237,7 +234,7 @@ def predict_gender_pronouns(
 
     indie_vars_list = indie_vars.split(',')
 
-    male_gendered_tokens, female_gendered_tokens = get_gendered_token_ids()
+    male_gendered_tokens, female_gendered_tokens = get_gendered_tokens()
 
     text_segments, num_preds = prepare_text_for_masking(
         input_text, mask_token, male_gendered_tokens + female_gendered_tokens, split_key)
@@ -301,13 +298,13 @@ description = """
 
 
 date_example = [
-    MODEL_NAMES[1],
+    MODEL_NAMES[4],
     '',  
     ', '.join(DATES),
     'DATE',
     "False",
     1,
-    'She was a teenager in DATE.'
+    'In DATE, she became an adult.'
 ]
 
 
@@ -318,7 +315,7 @@ place_example = [
     'PLACE',
     "False",
     1,
-    'She became an adult in PLACE.'
+    'She became a teen in PLACE.'
 ]