Views
No views yet
text-classification: Classify text into predefined medical conditions.1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3tokenizer = AutoTokenizer.from_pretrained("utkarshiitr/medicalchatbot")
4model = AutoModelForSequenceClassification.from_pretrained("utkarshiitr/medicalchatbot")
5
6inputs = tokenizer("fever, cough", return_tensors="pt")
7outputs = model(**inputs)