Update app.py

app.py

@@ -45,7 +45,7 @@ result = model(encoding)
 result
 st.write(result.logits)
 st.write(result.logits.size())
-st.write(result.logits.size(dim=0))
+st.write(result.logits.size(dim=1))
 
 if (result.logits.size(dim=0) >= 2):
     pad = (0, 1)