Update app.py

app.py

@@ -50,7 +50,7 @@ ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, aggregation
 def adr_predict(x):
     encoded_input = tokenizer(x, return_tensors='pt')
     output = model(**encoded_input)
-    scores = output[0][0].detach().numpy()
+    scores = output[0][0].detach()
     scores = torch.nn.functional.softmax(scores)
    
     shap_values = explainer([str(x).lower()])