jamesliounis commited on
Commit
907d23d
1 Parent(s): b38e03f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ You can use this model directly via the Hugging Face platform for tasks like abb
38
  from transformers import pipeline
39
 
40
  # Initialize the pipeline with the fine-tuned model
41
- clf = pipeline("text-classification", model="jamesliounis/MeDistilBERT")
42
 
43
  # Example text with medical abbreviation
44
  text = "Patient shows signs of CRF."
 
38
  from transformers import pipeline
39
 
40
  # Initialize the pipeline with the fine-tuned model
41
+ clf = pipeline("feature-extraction", model="jamesliounis/MeDistilBERT")
42
 
43
  # Example text with medical abbreviation
44
  text = "Patient shows signs of CRF."