Update

app.py

@@ -21,13 +21,16 @@ model_list = ['dumitrescustefan/bert-base-romanian-ner']
 st.sidebar.header("Select NER Model")
 model_checkpoint = st.sidebar.radio("", model_list)
 
-st.sidebar.write("This demo is based on RoNER: 'https://github.com/dumitrescustefan/roner'")
-st.sidebar.write("")
+
 
 
 st.sidebar.header("Select type of PERSON detection")
 named_persons_only_radio = st.sidebar.radio("", ('Proper nouns only', 'All nouns'))
 
+
+st.sidebar.write("This demo is based on RoNER: 'https://github.com/dumitrescustefan/roner'")
+st.sidebar.write("")
+st.sidebar.write("The NER corpus used is: 'https://github.com/dumitrescustefan/ronec'")
 st.sidebar.write("Types of entities detected: 'PERSON', 'ORG', 'GPE', 'LOC', 'NAT_REL_POL', 'EVENT', 'LANGUAGE', 'WORK_OF_ART', 'DATETIME', 'PERIOD', 'MONEY', 'QUANTITY', 'NUMERIC', 'ORDINAL', 'FACILITY'")
 
 st.subheader("Select Text Input Method")
@@ -108,8 +111,8 @@ if Run_Button == True:
     colors = {
         'PERSON': '#E08989',
         'ORG': '#54478C',
-        'GPE': '#2C699A',
-        'LOC': '#048BA8',
+        'GPE': '#4C89AA',
+        'LOC': '#34ABF8',
         'NAT_REL_POL': '#FADAD9',
         'EVENT': '#98654C',
         'LANGUAGE': '#F29E4C',