Sunmar commited on
Commit
a03159d
·
verified ·
1 Parent(s): 40b8f1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -11,7 +11,7 @@ def ner_models_result(address, models = fintuned_ner_models):
11
  ner_result_entities = []
12
  for model in models:
13
  pipe = pipeline("ner", model=f"{model}", aggregation_strategy="simple")
14
- ner_result_entities.append(pipe(address))
15
  return ner_result_entities
16
  st.title("Basic NER model testing")
17
  affiliation_address = st.text_input("Enter address")
@@ -19,4 +19,5 @@ if st.button("Print"):
19
  ner_results = ner_models_result(address = affiliation_address)
20
  for result in ner_results:
21
  st.write("-"*50)
22
- st.write(f"Result: {result}!")
 
 
11
  ner_result_entities = []
12
  for model in models:
13
  pipe = pipeline("ner", model=f"{model}", aggregation_strategy="simple")
14
+ ner_result_entities.append((model, pipe(address)))
15
  return ner_result_entities
16
  st.title("Basic NER model testing")
17
  affiliation_address = st.text_input("Enter address")
 
19
  ner_results = ner_models_result(address = affiliation_address)
20
  for result in ner_results:
21
  st.write("-"*50)
22
+ st.write(f"Model: {result[0]}")
23
+ st.write(f"Result: {result[1]}!")