lloydmeta
/

drug-bert

Text Classification

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

lloydmeta commited on Mar 17

Commit

0f6b2e4

•

1 Parent(s): f487940

Update README.md

Fix up usage sample

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -58,8 +58,8 @@ Actual, clinical diagnosis.
 ```python
 from transformers import pipeline
-condition_from_drug_review_classifier = pipeline("lloydmeta/drug-bert")
-text_sentiment = "I have been on quetiapine 50mg for 2 years now."
 condition_from_drug_review_classifier(text_sentiment)
 ```

 ```python
 from transformers import pipeline
+condition_from_drug_review_classifier = pipeline("text-classification", model = "lloydmeta/drug-bert")
+text_sentiment = "I have been taking ambien or zolphidem for almost 15 years."
 condition_from_drug_review_classifier(text_sentiment)
 ```