Update app.py

app.py

@@ -47,6 +47,7 @@ result
 if (result.logits.size(dim=1) < 2):
     pad = (0, 1)
     result = nn.functional.pad(result, pad, "constant", 0)
+    st.write(result)
     prediction = nn.functional.softmax(result, dim=-1)
 prediction = nn.functional.softmax(result.logits, dim=-1)
 prediction