CoolFace
Modelpublic

admesh/agentic-intent-classifier

sourceHugging Faceotherupdated 2d agoView on Hugging Face
2likes55downloads
inference_iab_retrieval.py21 linesDownload Raw Back to root
1import json2 3from iab_retrieval import predict_iab_content_retrieval4 5 6def predict(text: str):7    return predict_iab_content_retrieval(text)8 9 10examples = [11    "What is CRM software?",12    "How do I reset my password?",13    "Best project management tool for a startup team",14    "best laptop for coding",15]16 17if __name__ == "__main__":18    for text in examples:19        print(f"\nInput: {text}")20        print("Prediction:", json.dumps(predict(text), indent=2))21