Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -186,7 +186,6 @@ if st.button('확인'):
|
|
| 186 |
# logits = logits.detach().cpu().numpy()
|
| 187 |
|
| 188 |
st.write(logits.size())
|
| 189 |
-
st.wirte(len(logits[0]))
|
| 190 |
# # 단독 예측 시
|
| 191 |
# arg_idx = torch.argmax(logits, dim=1)
|
| 192 |
# print('arg_idx:', arg_idx)
|
|
|
|
| 186 |
# logits = logits.detach().cpu().numpy()
|
| 187 |
|
| 188 |
st.write(logits.size())
|
|
|
|
| 189 |
# # 단독 예측 시
|
| 190 |
# arg_idx = torch.argmax(logits, dim=1)
|
| 191 |
# print('arg_idx:', arg_idx)
|