Spaces:

ilushado
/

article_roberta_classifier

Runtime error

ilushado commited on Apr 23, 2023

Commit

59add47

1 Parent(s): 35145ea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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': ans.shape)
     st.markdown(f'{idx_to_tag[idx]}')
 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', ans.shape)
     st.markdown(f'{idx_to_tag[idx]}')
 if st.button("Посмотреть топ"):