Serega6678 commited on
Commit
6d96181
·
1 Parent(s): b411d6a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -2
README.md CHANGED
@@ -30,8 +30,16 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(
30
  'numind/entity-recognition-general-sota-v1'
31
  )
32
 
33
- text = "NuMind is an AI company based in Paris and USA."
34
- encoded_input = tokenizer(text, return_tensors='pt')
 
 
 
 
 
 
 
 
35
  output = model(**encoded_input)
36
 
37
  # for better quality
 
30
  'numind/entity-recognition-general-sota-v1'
31
  )
32
 
33
+ text = [
34
+ "NuMind is an AI company based in Paris and USA.",
35
+ "See other models from us on https://huggingface.co/numind"
36
+ ]
37
+ encoded_input = tokenizer(
38
+ text,
39
+ return_tensors='pt',
40
+ padding=True,
41
+ truncation=True
42
+ )
43
  output = model(**encoded_input)
44
 
45
  # for better quality