Keyurjotaniya007 commited on
Commit
4b34f38
·
verified ·
1 Parent(s): b0050c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- ner = pipeline('ner', model = "Keyurjotaniya007/xlm-roberta-base-xtreme-multilingual-ner-2.0")
5
 
6
  def merge_tokens(tokens):
7
  merged_tokens = []
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ ner = pipeline('ner', model = "Keyurjotaniya007/xlm-roberta-base-xtreme-multilingual-ner-2.0", grouped_entities=True)
5
 
6
  def merge_tokens(tokens):
7
  merged_tokens = []