Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|