Matt C commited on
Commit
6a356e0
1 Parent(s): 9a33247
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ result = model(batch)
19
  # get probabilities
20
  prediction = nn.functional.softmax(result.logits, dim=-1)
21
 
22
- print(prediction)
23
 
24
  #fig = px.bar(result, x="", y="", orientation='h')
25
  #fig.show()
 
19
  # get probabilities
20
  prediction = nn.functional.softmax(result.logits, dim=-1)
21
 
22
+ prediction
23
 
24
  #fig = px.bar(result, x="", y="", orientation='h')
25
  #fig.show()