texts = [
    "Congratulations! You've won a free ticket to Bahamas. Call now to claim your prize.",
    "Hey, are we still on for dinner tonight?"
]

for text in texts:
    print(classify_review(
        text, model, tokenizer, device,
        max_length=train_dataset.max_length
    ))

        __
        __