README.md

---
tags:
- text-classification
pipeline_tag: text-classification
---

# 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:

```python
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)