jschwaller commited on
Commit
a338a0a
·
verified ·
1 Parent(s): 2c638ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ model = AutoModelForSequenceClassification.from_pretrained("jschwaller/ADRv2024"
21
 
22
  # Build a pipeline object for predictions
23
  pred = transformers.pipeline("text-classification", model=model,
24
- tokenizer=tokenizer, return_all_scores=True)
25
 
26
  explainer = shap.Explainer(pred)
27
 
 
21
 
22
  # Build a pipeline object for predictions
23
  pred = transformers.pipeline("text-classification", model=model,
24
+ tokenizer=tokenizer, top_k=None)
25
 
26
  explainer = shap.Explainer(pred)
27