feat: added models choice

app.py

@@ -36,11 +36,13 @@ st.set_page_config(page_title="Named Entity Recognition")
 st.title("Named Entity Recognition")
 st.write("Type text into the text box and then press 'Predict' to get the named entities.")
 
-option = st.selectbox('Model', ("dslim/bert-base-NER", "flair/ner-english-fast", "Jean-Baptiste/camembert-ner"))
-st.write('Selected model:', option)
-
+option = st.selectbox('Select model',
+                      ("dslim/bert-base-NER",
+                       "dslim/bert-large-NER",
+                       "Davlan/bert-base-multilingual-cased-ner-hrl"))
 default_text = "Xbox v PlayStation: Giants clash over Call of Duty: Xbox owner Microsoft has hit back at claims its plan to buy the maker of Call of Duty may unfairly affect its rivals, including Sony, which owns PlayStation."
 text = st.text_area('Enter text here:', value=default_text)
+st.write('Model used for prediction:', option)
 submit = st.button('Predict')
 
 with st.spinner("Loading model..."):