fix
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ if st.button('Predict'):
|
|
25 |
# Predict label
|
26 |
|
27 |
with torch.no_grad():
|
28 |
-
|
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 |
|