Hugging Face's logo

Spaces:
nppmatt
/
milestone-2
Runtime error

App Files Community
nppmatt commited on
Commit
5933bb8
·
1 Parent(s): 67d3612

padding assign

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ encoding = tokenizer.encode(txt, return_tensors='pt')
43
  result = model(encoding)
44
  result
45
  pad = (0, 1)
46
- nn.functional.pad(result.logits, pad, "constant", 0)
47
  result
48
 
49
  # transform logit to get probabilities
 
43
  result = model(encoding)
44
  result
45
  pad = (0, 1)
46
+ result = nn.functional.pad(result, pad, "constant", 0)
47
  result
48
 
49
  # transform logit to get probabilities