MuntasirHossain commited on
Commit
18fbe0f
·
1 Parent(s): 9219f65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
 
4
  model_checkpoint = "MuntasirHossain/bert-finetuned-ner"
5
 
6
- model = pipeline("ner", model=model_checkpoint)
7
 
8
 
9
  # def ner(text):
@@ -21,7 +21,8 @@ def ner(text):
21
  description = "This AI model is trained to identify and classify named entities in unstructured text."
22
  title = "Named Entity Recognition"
23
  theme = "grass"
24
- examples=["Mount Everest is Earth's highest mountain, located in the Mahalangur Himal sub-range of the Himalayas. The China-Nepal border runs across it."]
 
25
 
26
  gr.Interface(fn=ner,
27
  inputs="textbox",
 
3
 
4
  model_checkpoint = "MuntasirHossain/bert-finetuned-ner"
5
 
6
+ model = pipeline("ner", model=model_checkpoint, aggregation_strategy="simple")
7
 
8
 
9
  # def ner(text):
 
21
  description = "This AI model is trained to identify and classify named entities in unstructured text."
22
  title = "Named Entity Recognition"
23
  theme = "grass"
24
+ examples=["Mount Everest is Earth's highest mountain, located in the Mahalangur Himal sub-range of the Himalayas. Edmund Hillary and Tenzing Norgay were the first climbers confirmed\
25
+ to have reached the summit of Mount Everest on May 29, 1953."]
26
 
27
  gr.Interface(fn=ner,
28
  inputs="textbox",