Hugging Face's logo

Spaces:
chuuhtetnaing
/
myanmar-text-segmentation-app
Running

App Files Community
chuuhtetnaing commited on
Commit
f140aab
·
1 Parent(s): 1c1c7e0

fixed the typo

Browse files
Files changed (1) hide show
  1. segmentation.py +1 -1
segmentation.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
 
6
  SEGMENTATION_MODEL_NAME = "chuuhtetnaing/myanmar-text-segmentation-model"
7
 
8
- classifier = pipeline("ner", model=SEGMENTATION_MODEL_NAME, grouped_entities=True)
9
 
10
 
11
  def reconstruct(tokens, labels):
 
5
 
6
  SEGMENTATION_MODEL_NAME = "chuuhtetnaing/myanmar-text-segmentation-model"
7
 
8
+ classifier = pipeline("token-classification", model=SEGMENTATION_MODEL_NAME, grouped_entities=True)
9
 
10
 
11
  def reconstruct(tokens, labels):