Update app.py

app.py

@@ -9,17 +9,20 @@ nlp = spacy.load("en_core_web_sm")
 
 def text_analysis(text):
     doc = nlp(text)
-    html = displacy.render(doc, style="dep", page=True)
-    html = (
+    dependency_parsing = displacy.render(doc, style="dep", page=True)
+    named_entity_recognition = displacy.render(doc, style="ent")
+    
+    visual1 = (
         "<div style='max-width:100%; max-height:360px; overflow:auto'>"
-        + html
+        + dependency_parsing
+        + "</div>"
+    )
+    
+    visual2 = (
+        "<div style='max-width:100%; max-height:360px; overflow:auto'>"
+        + named_entity_recognition
         + "</div>"
     )
-    pos_count = {
-        "char_count": len(text),
-        "token_count": 0,
-    }
-    pos_tokens = []
   
     rows = []
     for token in doc:
@@ -27,12 +30,14 @@ def text_analysis(text):
             token.shape_, token.is_alpha, token.is_stop))
   
     table = pd.DataFrame(rows, columns = ["TEXT", "LEMMA","POS","TAG","DEP","SHAPE","ALPHA","STOP"])
-    return table, html
+
+    return table, visual1, visual2
+
 
 demo = gr.Interface(
     text_analysis,
     gr.Textbox(placeholder="Enter sentence here..."),
-    [gr.Dataframe(), "html"],
+    [gr.Dataframe(), "html", "html"],
     examples=[
         ["Data Science Dojo is the leading platform providing training in data science, data analytics, and machine learning."],
         ["It's the best time to execute the plan."],