Update README.md

README.md

@@ -24,4 +24,10 @@ 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")
+
+params = {"decoder_start_token_id":0,"early_stopping":False,"no_repeat_ngram_size":0,"length_penalty": 0,"num_beams":20,"use_cache":True}
+
+out_id = model.generate(inputs["input_ids"], attention_mask = inputs["attention_mask"], max_length=max_target_length, **params)