Hugging Face's logo

fractalego
/
fewrel-zero-shot

Question Answering
Transformers
PyTorch
bert
Inference Endpoints
Model card Files Community
1
Fractalego commited on
Commit
509e315
•
1 Parent(s): 664e452

updated README

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -8,10 +8,10 @@ $ pip install zero-shot-re
8
 
9
  ## Run the Extractor
10
  ```python
11
- from transformers import AutoModel, AutoTokenizer
12
- from zero-shot-re import RelationExtractor
13
 
14
- model = AutoModel.from_pretrained("fractalego/fewrel-zero-shot")
15
  tokenizer = AutoTokenizer.from_pretrained("fractalego/fewrel-zero-shot")
16
 
17
  relations = ['noble title', 'founding date', 'occupation of a person']
8
 
9
  ## Run the Extractor
10
  ```python
11
+ from transformers import AutoTokenizer
12
+ from zero_shot_re import RelTaggerModel, RelationExtractor
13
 
14
+ model = RelTaggerModel.from_pretrained("fractalego/fewrel-zero-shot")
15
  tokenizer = AutoTokenizer.from_pretrained("fractalego/fewrel-zero-shot")
16
 
17
  relations = ['noble title', 'founding date', 'occupation of a person']