Update app.py

app.py

@@ -86,6 +86,7 @@ st.markdown("### Hello, world!")
 
 title = st.text_area("Title HERE")
 abstract = st.text_area("Abstract HERE")
+ans = None
 
 
 if st.button('Предположить'):
@@ -108,6 +109,35 @@ if st.button('Предположить'):
     idx = torch.nn.functional.softmax(ans[0], dim=0).argmax().item()
     st.markdown(f'{idx_to_tag[idx]}')
 
+if st.button("Посмотреть топ"):
+    if not ans:
+        text = title + " : " + abstract
+        inputs = tokenizer.encode_plus(
+                    text,
+                    None,
+                    add_special_tokens=True,
+                    max_length=256,
+                    pad_to_max_length=True,
+                    return_token_type_ids=True
+                )
+        
+        
+        ids = torch.Tensor(inputs['input_ids']).long()
+        mask = torch.Tensor(inputs['attention_mask']).long()
+        token_type_ids = torch.Tensor(inputs['token_type_ids']).long()
+        
+        ans = model(ids.unsqueeze(0), mask.unsqueeze(0), token_type_ids.unsqueeze(0))
+    elems = [el.item() for el in ans[0].argsort(descending=True)[:10]]
+    probs = ans[0].softmax(dim=0)
+    str = ''
+    for el in elems:
+        str += el + " : " + probs[el] + "\n"
+
+    st.markdown(str)
+
+    
+    
+