nppmatt commited on
Commit
57147ca
·
1 Parent(s): ded950b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -43,8 +43,9 @@ encoding = tokenizer.encode(txt, return_tensors='pt')
43
  result = model(encoding)
44
 
45
  result
46
- result.logits.size()
47
- result.logits.size(dim=0)
 
48
 
49
  if (result.logits.size(dim=0) >= 2):
50
  pad = (0, 1)
 
43
  result = model(encoding)
44
 
45
  result
46
+ st.write(result.logits))
47
+ st.write(result.logits.size())
48
+ st.write(result.logits.size(dim=0))
49
 
50
  if (result.logits.size(dim=0) >= 2):
51
  pad = (0, 1)