File size: 930 Bytes
0584798 869e285 0584798 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | import argparse
import json
try:
from .model_runtime import get_head # type: ignore
except ImportError:
from model_runtime import get_head
def predict(text: str, confidence_threshold: float | None = None):
return get_head("intent_type").predict(text, confidence_threshold=confidence_threshold)
examples = [
"What is CRM?",
"Best CRM for small teams",
"HubSpot vs Zoho CRM",
"Tell me more",
]
def main():
parser = argparse.ArgumentParser(description="Run intent_type inference for one query or the built-in examples.")
parser.add_argument("text", nargs="?", help="Optional query text to classify.")
args = parser.parse_args()
if args.text:
print(json.dumps(predict(args.text), indent=2))
return
for text in examples:
print(f"\nInput: {text}")
print("Prediction:", json.dumps(predict(text), indent=2))
if __name__ == "__main__":
main()
|