Update app.py

app.py

@@ -185,18 +185,22 @@ with col1:
         table["Entity"].append(key)
         table["Class"].append(ent_bc[key])
     trans_df = pd.DataFrame(table)
-    st.subheader("DISEASE")
-    genEntities(trans_df, 'DISEASE')
-    st.subheader("CHEMICAL")
-    genEntities(trans_df, 'CHEMICAL')
+#    st.subheader("DISEASE")
+#    genEntities(trans_df, 'DISEASE')
+#    st.subheader("CHEMICAL")
+#    genEntities(trans_df, 'CHEMICAL')
     #st.table(trans_df)    
     
 with col2:
     st.button('NER')
-#    doc = nlp(str(original_text2))
-#    colors = { "DISEASE": "pink","CHEMICAL": "orange"}
-#    options = {"ents": [ "DISEASE", "CHEMICAL"],"colors": colors}
-#    ent_html = displacy.render(doc, style="ent", options=options)
+    st.subheader("DISEASE")
+    genEntities(trans_df, 'DISEASE')
+    st.subheader("CHEMICAL")
+    genEntities(trans_df, 'CHEMICAL')
+    #st.table(trans_df)   
+    st.subheader("NER")
     st.markdown(ent_html, unsafe_allow_html=True) 
-    #st.write(doc.ents)
+  
+
+