Views
No views yet
1from transformers import T5ForConditionalGeneration, T5Tokenizer
2
3model = T5ForConditionalGeneration.from_pretrained("pdarleyjr/iplc-t5-model")
4tokenizer = T5Tokenizer.from_pretrained("pdarleyjr/iplc-t5-model")
5
6text = "summarize: evaluation type: initial. primary diagnosis: F84.0. severity: mild. primary language: english"
7input_ids = tokenizer.encode(text, return_tensors="pt", max_length=512, truncation=True)
8
9outputs = model.generate(
10 input_ids,
11 max_length=256,
12 num_beams=4,
13 no_repeat_ngram_size=3,
14 length_penalty=2.0,
15 early_stopping=True
16)
17
18summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
19print(summary)