Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -47,6 +47,7 @@ result
|
|
47 |
if (result.logits.size(dim=1) < 2):
|
48 |
pad = (0, 1)
|
49 |
result = nn.functional.pad(result, pad, "constant", 0)
|
|
|
50 |
prediction = nn.functional.softmax(result, dim=-1)
|
51 |
prediction = nn.functional.softmax(result.logits, dim=-1)
|
52 |
prediction
|
|
|
47 |
if (result.logits.size(dim=1) < 2):
|
48 |
pad = (0, 1)
|
49 |
result = nn.functional.pad(result, pad, "constant", 0)
|
50 |
+
st.write(result)
|
51 |
prediction = nn.functional.softmax(result, dim=-1)
|
52 |
prediction = nn.functional.softmax(result.logits, dim=-1)
|
53 |
prediction
|