Update README.md

README.md

@@ -87,6 +87,19 @@ MODEL_CKPT = 'mrm8488/t5-base-iterater'
 tokenizer = T5TokenizerFast.from_pretrained(MODEL_CKPT)
 model = T5ForConditionalGeneration.from_pretrained(MODEL_CKPT)
 
+def predict(intent, text):
+   input_text =  f"<{intent}>  {text}"
+   features = tokenizer([input_text], return_tensors='pt')
+   output = model.generate(input_ids=features['input_ids'], 
+               attention_mask=features['attention_mask'], max_length=128, num_beams=8)
+   return tokenizer.decode(output[0], skip_special_tokens=True)
+   
+text = "Delay-based schemes have the potential to resolve this last packet problem by scheduling the link based on the delay for the packet has encountered."
+intent = "clarity"
+
+predict(intent, text)
+# Delay-based schemes have the potential to resolve this last packet problem by scheduling the link based on the delay the packet has encountered.
+
 ```
 
 ### Framework versions