Quick hack as realized return type of `mask_filled_text` based on how many MASKs in text

app.py

@@ -219,6 +219,9 @@ def predict_gender_pronouns(
             model = models[(bert_like,)]
 
             mask_filled_text = model(target_text_for_bert)
+            # Quick hack as realized return type based on how many MASKs in text.
+            if type(mask_filled_text[0]) is not list:
+                mask_filled_text = [mask_filled_text]
 
             female_pronouns = [
                 1 if pronoun[0]["token_str"] in female_gendered_tokens else 0