Update app.py

app.py

@@ -185,7 +185,8 @@ if st.button('확인'):
         # Move logits and labels to CPU
 #        logits = logits.detach().cpu().numpy()
 
-    st.write(logits)
+    st.write(logits.size())
+    st.wirte(len(logits[0]))
     # # 단독 예측 시
     # arg_idx = torch.argmax(logits, dim=1)
     # print('arg_idx:', arg_idx)