Downloads · 30 days
7
13% of all-time downloads
medicalchatbot_2
utkarshiitr/medicalchatbot_2
medicalchatbot_2 is a text classification model from utkarshiitr. Use it when you need a label for a piece of text. It is set up for transformers.
This model is trained to classify medical conditions based on input symptoms and provide appropriate recommendations.
All-time downloads
54
Public
Parameters
110M
438 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors438 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Jun 16, 2024
- Updated
- Jun 16, 2024
- SHA
- 6b6d074e
Try a prompt
Model card
From the Hugging Face model README
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)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Created
- Jun 16, 2024
- Updated
- Jun 16, 2024