Views
No views yet
1pip install transformers tensorflow
2
3from transformers import pipeline, AutoTokenizer, TFAutoModelForSequenceClassification
4
5model_name = "MadeForMoney/triage-text-classifier"
6
7tokenizer = AutoTokenizer.from_pretrained(model_name)
8model = TFAutoModelForSequenceClassification.from_pretrained(model_name)
9
10classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
11
12result = classifier("Patient reports severe headache and dizziness")
13print(result)