Medical Diagnosis Chatbot
This model is trained to classify medical conditions based on input symptoms and provide appropriate recommendations.
Supported Tasks
This model supports the following tasks:
text-classification: Classify text into predefined medical conditions.
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)
- Downloads last month
- 20
This model does not have enough activity to be deployed to Inference API (serverless) yet.
Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.