mavinsao commited on
Commit
ae12529
·
verified ·
1 Parent(s): 2b716d2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ if st.button('Predict'):
25
  # Predict label
26
 
27
  with torch.no_grad():
28
- logits = model(sentence).logits
29
 
30
  predicted_class_ids = torch.arange(0, logits.shape[-1])[torch.sigmoid(logits).squeeze(dim=0) > 0.5]
31
 
 
25
  # Predict label
26
 
27
  with torch.no_grad():
28
+ logits = model(sentence).logits
29
 
30
  predicted_class_ids = torch.arange(0, logits.shape[-1])[torch.sigmoid(logits).squeeze(dim=0) > 0.5]
31