Update app.py

app.py

@@ -2,9 +2,21 @@
 from transformers import pipeline
 import streamlit as st
 
-pipe = pipeline("ner", model="SIRIS-Lab/affilgood-NER-multilingual", aggregation_strategy="simple")
-
+# dslim/bert-base-NER
+# SIRIS-Lab/affilgood-NER-multilingual
+# FacebookAI/xlm-roberta-large-finetuned-conll03-english
+#
+fintuned_ner_models = ["dslim/bert-base-NER", "SIRIS-Lab/affilgood-NER-multilingual", "FacebookAI/xlm-roberta-large-finetuned-conll03-english"]
+def ner_models_result(address, models = fintuned_ner_models):
+    ner_result_entities = []
+    for model in models:
+        pipe = pipeline("ner", model=f"{model}", aggregation_strategy="simple")
+        ner_result_entities.append(pipe(address))
+    return ner_result_entities
 st.title("Basic NER model testing")
 affiliation_address = st.text_input("Enter address")
 if st.button("Print"):
-    st.write(f"Result: {pipe(affiliation_address)}!")
+    ner_results = ner_models_result(address = affiliation_address)
+    for result in ner_results:
+        st.write("-"*50)
+        st.write(f"Result: {result}!")