How to Get Started With the Model
Example of single-label classification: โโ
!pip install -U transformers
from transformers import pipeline
classifier = pipeline('text-classification', model= 'piyush23111991/bert-base-uncased-symptom-to-disease-prediction')
classifier(['I am experiencing too many cramps in the last couple of days. I think something is not right. I believe there is a small bruise on my calves, but I am not sure about it.'])
#[{'label': 'Varicose Veins', 'score': 0.5619356632232666}]
- Downloads last month
- 14
Model tree for piyush23111991/bert-base-uncased-symptom-to-disease-prediction
Base model
google-bert/bert-base-uncased