Hugging Face's logo

knowledgator
/
t5-for-ie

Text Generation
Transformers
PyTorch
English
t5
text2text-generation
information extraction
entity linking
named entity recogntion
relation extraction
text-to-text generation
text-generation-inference
Model card Files
xet
Community
1
Ihor commited on
Commit
6ac3d8e
·
verified ·
1 Parent(s): 6f8d7e8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -75,9 +75,10 @@ labels=[
75
  ]
76
 
77
  classifier = TextClassifier(
78
- labels=None,
79
  model=model,
80
  tokenizer=tokenizer,
 
81
  )
82
  classifier.initialize_labels_trie(labels)
83
 
 
75
  ]
76
 
77
  classifier = TextClassifier(
78
+ labels=['default'],
79
  model=model,
80
  tokenizer=tokenizer,
81
+ device='cuda:0' #if cuda
82
  )
83
  classifier.initialize_labels_trie(labels)
84