Update app.py

app.py

@@ -111,7 +111,7 @@ def genEntities(ann, entity):
               if entity in entities:
                              ent = list(trans_df[trans_df['Class']==entity]['Entity'].unique())
                              entlist = ",".join(ent)
-                             st.markdown(f'<p style="color:{ent_col[entity]};font-size:20px;">{entlist}</p>', unsafe_allow_html=True)
+                             st.markdown(f'<p style="color:{ent_col[entity]};font-size:16px;">{entlist}</p>', unsafe_allow_html=True)
                              #for i in ent:
                                            #st.markdown(f'<p style="color:{ent_col[entity]};font-size:20px;">{i}</p>', unsafe_allow_html=True)
                                           
@@ -185,12 +185,7 @@ 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.table(trans_df)    
-    
+
 with col2:
     st.button('NER')
     st.markdown('**DISEASE**')
@@ -200,7 +195,3 @@ with col2:
     #st.table(trans_df)   
     st.markdown('**NER**')
     st.markdown(ent_html, unsafe_allow_html=True) 
-  
-
-    
-