Downloads · 30 days
6
27% of all-time downloads
medicalchatbot
utkarshiitr/medicalchatbot
medicalchatbot is a machine learning model from utkarshiitr. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This model supports the following tasks: - text-classification: Classify text into predefined categories.
All-time downloads
22
Public
Parameters
110M
877 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors438 MB · 100%
At a glance
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Jun 16, 2024
- Updated
- Jun 16, 2024
- SHA
- e9a07275
Model card
From the Hugging Face model README
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)
- Library
- transformers
- Type
- bert
- Created
- Jun 16, 2024
- Updated
- Jun 16, 2024