Update app.py

app.py

@@ -87,7 +87,7 @@ if st.button('Предположить'):
     
     ans = model(ids.unsqueeze(0), mask.unsqueeze(0), token_type_ids.unsqueeze(0))
     idx = torch.nn.functional.softmax(ans[0], dim=0).argmax().item()
-    print('ANSLEN': len(ans))
+    print('ANSLEN': ans.shape)
     st.markdown(f'{idx_to_tag[idx]}')
 
 if st.button("Посмотреть топ"):