fanxiao
/

CGRE_CNDBPedia-Generative-Relation-Extraction

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

fanxiao commited on Apr 17, 2023

Commit

f4a105c

•

1 Parent(s): a58d18d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ tokenizer_kwargs = {
 tokenizer = BertTokenizer.from_pretrained(model_name, **tokenizer_kwargs)
-model = BartForConditionalGeneration.from_pretrained('./CGRE_CNDBPedia-Generative-Relation-Extraction',local_files_only=True)
 inputs = tokenizer(sent, max_length=max_source_length, padding="max_length", truncation=True, return_tensors="pt")

 tokenizer = BertTokenizer.from_pretrained(model_name, **tokenizer_kwargs)
+model = BartForConditionalGeneration.from_pretrained('./CGRE_CNDBPedia-Generative-Relation-Extraction')
 inputs = tokenizer(sent, max_length=max_source_length, padding="max_length", truncation=True, return_tensors="pt")