gretelai/symptom_to_diagnosis
Viewer • Updated • 1.07k • 820 • 48
How to use kechemale/eng-am-symptom-ner with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="kechemale/eng-am-symptom-ner") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("kechemale/eng-am-symptom-ner")
model = AutoModelForTokenClassification.from_pretrained("kechemale/eng-am-symptom-ner")This model is a fine-tuned XLM-R-base model for extracting symptoms from patient generated texts in English and Amharic. It was developed as part of an MSc thesis at the Technical University of Munich, aiming to support AI-powered symptom extraction platforms in multilingual healthcare settings.
The model performs named entity recognition (NER) to identify symptom mentions in unstructured texts using the BIO tagging scheme.
Intended uses:
Base model
FacebookAI/xlm-roberta-base