--- license: mit language: - en library_name: transformers pipeline_tag: text-classification tags: - bert-base-uncased - disease - medical widget: - text: "I am having itching, skin rash, and nodal skin eruptions" example_title: "Fungal infection example" - text: "I feel like vomiting, breathlessness, and sweating" example_title: "Heart Attack example" - text: "I am feeling fatigue, weight loss, restlessness and also lethargy." example_title: "Diabetes example" --- 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. Labels: 0 - Fungal infection 1 - Diabetes 2 - Heart attack Will add more diseases in coming days