Update app.py

app.py

@@ -14,7 +14,7 @@ def Predict(model, tokenizer, text):
   res = tokenizer(text, padding=True, truncation=True, return_tensors="pt", max_length=512)
   res = model(**res)
   logits = res.logits.softmax(dim=1)
-  logits = logits.numpy()[0]#.cpu().detach().numpy()[0]
+  logits = logits.detach().numpy()[0]#.cpu().detach().numpy()[0]
   return logits
   
 def Print(logits, dictionary):