Spaces:

nppmatt
/

milestone-2

Runtime error

nppmatt commited on Apr 10, 2023

Commit

98eea47

1 Parent(s): 67e928c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 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