File size: 462 Bytes
0584798 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | import json
from iab_retrieval import predict_iab_content_retrieval
def predict(text: str):
return predict_iab_content_retrieval(text)
examples = [
"What is CRM software?",
"How do I reset my password?",
"Best project management tool for a startup team",
"best laptop for coding",
]
if __name__ == "__main__":
for text in examples:
print(f"\nInput: {text}")
print("Prediction:", json.dumps(predict(text), indent=2))
|