Edit model card

The objective is to develop a symptom-to-disease classification model for a natural language chatbot. This model takes input text such as "I am feeling vomiting, breathlessness, and sweating" and accurately identifies the associated disease (2 - 'Heart attack').

In essence, the chatbot's purpose is to analyze users' symptoms and provide relevant disease predictions in real-time conversation.

0 - Fungal infection
1 - Diabetes
2 - Heart attack

Will add more diseases in coming days

Downloads last month
Model size
109M params
Tensor type