Update OpenMed.py

OpenMed.py

@@ -1,6 +1,21 @@
+
 from transformers import pipeline
 
-ner_pipeline = pipeline("token-classification", model="OpenMed/OpenMed-NER-PharmaDetect-SuperClinical-434M", aggregation_strategy="simple")
-text = "Patient prescribed 10mg aspirin for hypertension."
-entities = ner_pipeline(text)
+# Load the model and tokenizer
+# Model: https://huggingface.co/OpenMed/OpenMed-NER-GenomeDetect-BigMed-278M
+model_name = "OpenMed/OpenMed-NER-GenomeDetect-BigMed-278M"
+
+# Create a pipeline
+medical_ner_pipeline = pipeline(
+    model=model_name,
+    aggregation_strategy="simple"
+)
+
+# Example usage
+text = "The EGFR gene mutation was identified in lung cancer patients."
+entities = medical_ner_pipeline(text)
+
 print(entities)
+
+token = entities[0]
+print(text[token["start"] : token["end"]])