File size: 736 Bytes
0584798 2b0e686 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 | 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("decision_phase").predict(text, confidence_threshold=confidence_threshold)
examples = [
"What is CRM software?",
"What are some CRM options for startups?",
"HubSpot vs Zoho for a small team",
"Which CRM should I buy for a 3-person startup?",
"Start my free trial",
"How do I set up my new CRM?",
"I cannot log into my account",
]
if __name__ == "__main__":
for text in examples:
print(f"\nInput: {text}")
print("Prediction:", json.dumps(predict(text), indent=2))
|