nppmatt commited on
Commit
50ad213
1 Parent(s): 58f6b64

more println debugging

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -40,6 +40,7 @@ model = AutoModelForSequenceClassification.from_pretrained(modelPath)
40
  # RoBERTA: [0]: neutral, [1]: toxic
41
  encoding = tokenizer.encode(txt, return_tensors='pt')
42
  result = model(encoding)
 
43
 
44
  # transform logit to get probabilities
45
  prediction = nn.functional.softmax(result.logits, dim=-1)
 
40
  # RoBERTA: [0]: neutral, [1]: toxic
41
  encoding = tokenizer.encode(txt, return_tensors='pt')
42
  result = model(encoding)
43
+ result
44
 
45
  # transform logit to get probabilities
46
  prediction = nn.functional.softmax(result.logits, dim=-1)