Create README.md

README.md

@@ -0,0 +1,9 @@
+from transformers import T5ForConditionalGeneration, T5Tokenizer
+import torch
+model = T5ForConditionalGeneration.from_pretrained("sana-ngu/HaT5")
+tokenizer = T5Tokenizer.from_pretrained("t5-base") # use the source tokenizer because T5 finetuned tokenizer breaks
+tokenizer.pad_token = tokenizer.eos_token
+input_ids = tokenizer("Old lions in the wild lay down and die with dignity when they can't hunt anymore. If a government is having 'teething problems' handling aid supplies one full year into a pandemic, maybe it should take a cue and get the fuck out of the way? ", padding=True, truncation=True, return_tensors='pt').input_ids
+outputs = model.generate(input_ids)
+pred = tokenizer.decode(outputs[0], skip_special_tokens=True)
+print(pred)