utkarshiitr/medicalchatbot

Supported Tasks

This model supports the following tasks:

• text-classification: Classify text into predefined categories.

Usage

Here is how you can use the model for text classification:

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("utkarshiitr/medicalchatbot")
model = AutoModelForSequenceClassification.from_pretrained("utkarshiitr/medicalchatbot")

inputs = tokenizer("fever, cough", return_tensors="pt")
outputs = model(**inputs)