VibeCheck-AI / predict.py
ajm19826's picture
Create predict.py
bf53e60 verified
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")