MiyunKim commited on
Commit
da2dfa0
·
verified ·
1 Parent(s): ac22e5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -6
app.py CHANGED
@@ -41,12 +41,6 @@ explainer = shap.Explainer(pred)
41
  # score_1sym = x['score']
42
  # return round(score_1sym,3)
43
 
44
- ner_tokenizer = AutoTokenizer.from_pretrained("d4data/biomedical-ner-all")
45
- ner_model = AutoModelForTokenClassification.from_pretrained("d4data/biomedical-ner-all")
46
-
47
- ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, aggregation_strategy="simple") # pass device=0 if using gpu
48
- #
49
-
50
  def adr_predict(x):
51
  encoded_input = tokenizer(x, return_tensors='pt')
52
  output = model(**encoded_input)
 
41
  # score_1sym = x['score']
42
  # return round(score_1sym,3)
43
 
 
 
 
 
 
 
44
  def adr_predict(x):
45
  encoded_input = tokenizer(x, return_tensors='pt')
46
  output = model(**encoded_input)