Create test.py

Model test file.

model_trainer/test.py

@@ -0,0 +1,11 @@
+from transformers import PegasusForConditionalGeneration, PegasusTokenizer
+
+model_name = "your-username/pegasus-samsum-model"
+model = PegasusForConditionalGeneration.from_pretrained(model_name)
+tokenizer = PegasusTokenizer.from_pretrained(f"{model_name}/tokenizer")
+
+# Test the model
+text = "Your input text here"
+tokens = tokenizer(text, truncation=True, padding="longest", return_tensors="pt")
+summary = model.generate(**tokens)
+print(tokenizer.decode(summary[0], skip_special_tokens=True))