Update app.py

app.py

@@ -19,7 +19,7 @@ if st.button("Highlight Geo-Entities"):
         # Highlight geo-entities
         highlighted_text = user_input
         for ent in reversed(doc.ents):
-            if ent.label_ in ["GPE", "LOC"]:  # GPE = Geopolitical Entity, LOC = Location
+            if ent.label_ in ['FAC', 'ORG', 'LOC', 'GPE']:  # GPE = Geopolitical Entity, LOC = Location
                 highlighted_text = (
                     highlighted_text[:ent.start_char] +
                     f"**:green[{ent.text}]**" +