Update app.py

app.py

@@ -96,7 +96,7 @@ def create_mask_dict(entities, additional_masks=None):
     entity_counters = {}
     
     for entity in entities:
-        if entity['entity_group'] not in ['CARDINAL', 'EVENT', 'PERCENT', 'QUANTITY']:
+        if entity['entity_group'] not in ['CARDINAL', 'EVENT', 'PERCENT', 'QUANTITY', 'DATE', 'TITLE', 'WORK_OF_ART']:
             if entity['word'] not in mask_dict:  # Corrected indentation
                 if entity['entity_group'] not in entity_counters:
                     entity_counters[entity['entity_group']] = 1