Update app.py

app.py

@@ -43,14 +43,9 @@ encoding = tokenizer.encode(txt, return_tensors='pt')
 result = model(encoding)
 
 result
-st.write(result.logits)
-st.write(result.logits.size())
-st.write(result.logits.size(dim=1))
-
-if (result.logits.size(dim=0) >= 2):
+if (result.logits.size(dim=1) >= 2):
     pad = (0, 1)
     result = nn.functional.pad(result, pad, "constant", 0)
-
 result
 
 # transform logit to get probabilities