predict.py

from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="./model",
    tokenizer="./model"
)

while True:
    text = input("Say something > ")
    if text.lower() in ["exit", "quit"]:
        break

    result = classifier(text)[0]
    print(f"✨ Vibe: {result['label']} ({result['score']:.2f})\n")