Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,3 @@
|
|
| 1 |
-
from asyncore import write
|
| 2 |
-
from pickletools import stringnl
|
| 3 |
import streamlit as st
|
| 4 |
import pandas as pd
|
| 5 |
|
|
@@ -97,7 +95,7 @@ if st.button('확인'):
|
|
| 97 |
# str_ans = loc_tbl['항목명'][loc_tbl['코드'] == num_ans].values
|
| 98 |
|
| 99 |
# 상위 k번째까지 예측 시
|
| 100 |
-
k =
|
| 101 |
topk_idx = torch.topk(logits.flatten(), k).indices
|
| 102 |
|
| 103 |
num_ans_topk = label_tbl[topk_idx]
|
|
|
|
|
|
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
import pandas as pd
|
| 3 |
|
|
|
|
| 95 |
# str_ans = loc_tbl['항목명'][loc_tbl['코드'] == num_ans].values
|
| 96 |
|
| 97 |
# 상위 k번째까지 예측 시
|
| 98 |
+
k = 10
|
| 99 |
topk_idx = torch.topk(logits.flatten(), k).indices
|
| 100 |
|
| 101 |
num_ans_topk = label_tbl[topk_idx]
|