Update app.py

app.py

@@ -41,9 +41,11 @@ model = AutoModelForSequenceClassification.from_pretrained(modelPath)
 # RoBERTA: [0]: neutral, [1]: toxic
 encoding = tokenizer.encode(txt, return_tensors='pt')
 result = model(encoding)
+
 result
-pad = (0, 1)
-result = nn.functional.pad(result.logits, pad, "constant", 0)
+if (result.logits.size(dim=0) >= 2):
+    pad = (0, 1)
+    result = nn.functional.pad(result, pad, "constant", 0)
 result
 
 # transform logit to get probabilities