Update app.py

app.py

@@ -46,9 +46,8 @@ result
 # transform logit to get probabilities
 if (result.logits.size(dim=1) < 2):
     pad = (0, 1)
-    result = nn.functional.pad(result.logits, pad, "constant", 0)
+    result.logits = nn.functional.pad(result.logits, pad, "constant", 0)
     st.write(result)
-    prediction = nn.functional.softmax(result, dim=-1)
 prediction = nn.functional.softmax(result.logits, dim=-1)
 prediction
 #neutralProb = prediction.data[0][neutralIndex]