Update model_functions.py

model_functions.py

@@ -86,6 +86,15 @@ def get_NER(text, pipe):
     filtered_results = sorted(unique_entities.values(), key=lambda x: x['start'])
     # Format the results for a table display
     formatted_results = [[ent['word'], ent['entity_group']] for ent in filtered_results]
-
-    return formatted_results
+    filtered_results = []
+    for entity in formatted_results:
+        if entity[1] == 'ORG':
+            # Split the 'word' by spaces and count the number of words
+            if len(entity[0].split()) <= 2:
+                filtered_results.append(entity)
+        else:
+            # Add non-ORG entities without filtering
+            filtered_results.append(entity)
+
+    return filtered_results