IntentSnap-AI / predict.py
ajm19826's picture
Create predict.py
f81688c verified
from transformers import pipeline
classifier = pipeline(
"text-classification",
model="./model",
tokenizer="./model"
)
print("🤖 IntentSnap-AI (type 'exit' to quit)\n")
while True:
text = input("> ")
if text.lower() in ["exit", "quit"]:
break
result = classifier(text)[0]
print(f"🎯 Intent: {result['label']} ({result['score']:.2f})\n")