display entities

app.py

@@ -16,6 +16,7 @@ import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt
 import seaborn as sns
+from spacy import displacy
 from nltk import word_tokenize
 from nltk.probability import FreqDist
 from matplotlib import pyplot as plt
@@ -100,6 +101,12 @@ with st.form('model_prediction'):
                 date = "None found" if len(date)<1 else date
                 st.write("Location entities: " + loc)
                 st.write("Date entities: " + date)
+                
+                #Display entities
+                st.write("All Entities-")
+                ent_html = displacy.render(n_text, style="ent", jupyter=False)
+                # Display the entity visualization in the browser:
+                st.markdown(ent_html, unsafe_allow_html=True)
 
 st.header("Predict on test data")
 with st.form('model_test_prediction'):