wibberlet commited on
Commit
c692027
1 Parent(s): d16fbd9

Update NamedEntity.py

Browse files
Files changed (1) hide show
  1. NamedEntity.py +0 -7
NamedEntity.py CHANGED
@@ -32,16 +32,9 @@ class NER:
32
  The Constructor for the Named Entity Recognition class.
33
  :param text_to_analyse: The text in which to find named entities.
34
  """
35
- if text_to_analyse is None or len(text_to_analyse.strip()) == 0:
36
- raise ValueError("text_to_analyse must not be empty and must be set to a valid string value")
37
-
38
  self.tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
39
- if self.tokenizer is None:
40
- raise ValueError("Unable to load tokenizer from DSLIM BERT model")
41
 
42
  self.model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
43
- if self.model is None:
44
- raise ValueError("Unable to load model from DSLIM BERT model")
45
 
46
  self.nlp = pipeline("ner", model=self.model, tokenizer=self.tokenizer, grouped_entities=True)
47
  if self.nlp is None:
 
32
  The Constructor for the Named Entity Recognition class.
33
  :param text_to_analyse: The text in which to find named entities.
34
  """
 
 
 
35
  self.tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
 
 
36
 
37
  self.model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
 
 
38
 
39
  self.nlp = pipeline("ner", model=self.model, tokenizer=self.tokenizer, grouped_entities=True)
40
  if self.nlp is None: