Spaces:
Sleeping
Sleeping
jschwaller
commited on
Update app.py
Browse files
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,
|
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 |
|