Update app.py

app.py

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