Update app.py

app.py

@@ -23,8 +23,7 @@ model = AutoModelForSequenceClassification.from_pretrained("paragon-analytics/AD
 
 # Build a pipeline object for predictions
 # Note: return_all_scores is deprecated, use top_k=None instead
-pred = transformers.pipeline("text-classification", model=model,
-                              tokenizer=tokenizer, return_all_scores=True) # This will likely issue a warning or error in newer versions.
+pred = transformers.pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=None, device=device) # Added device=device for consistency
 
 # SHAP explainer
 # Check SHAP documentation for potential changes in Explainer initialization if issues arise.