Update app.py

app.py

@@ -110,12 +110,13 @@ if st.button("Посмотреть топ"):
         
         ans = model(ids.unsqueeze(0), mask.unsqueeze(0), token_type_ids.unsqueeze(0))
     elems = [el.item() for el in ans[0].argsort(descending=True)]
-    probs = ans[0].softmax(dim=0)
+    probs = ans[0].softmax(dim=0).detach().numpy()
     str_ans = ''
     current_prob = 0
     current_elems = []
     current_probs = []
     idx = 0
+    
     while current_prob < 0.95 and idx < len(elems):
         current_elems.append(idx_to_tag[elems[idx]])
         current_probs.append(probs[elems[idx]])