Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -184,6 +184,7 @@ if st.button('확인'):
|
|
| 184 |
|
| 185 |
# Move logits and labels to CPU
|
| 186 |
# logits = logits.detach().cpu().numpy()
|
|
|
|
| 187 |
pred_m = torch.nn.Softmax(dim=1)
|
| 188 |
pred_ = pred_m(logits)
|
| 189 |
# st.write(logits.size())
|
|
@@ -231,7 +232,7 @@ if st.button('확인'):
|
|
| 231 |
percent_ans_topk_list = []
|
| 232 |
for i in range(k):
|
| 233 |
str_ans_topk_list.append(str_ans_topk[i].iloc[0])
|
| 234 |
-
percent_ans_topk_list.append(percent_ans_topk[i])
|
| 235 |
|
| 236 |
# print(str_ans_topk_list)
|
| 237 |
|
|
|
|
| 184 |
|
| 185 |
# Move logits and labels to CPU
|
| 186 |
# logits = logits.detach().cpu().numpy()
|
| 187 |
+
|
| 188 |
pred_m = torch.nn.Softmax(dim=1)
|
| 189 |
pred_ = pred_m(logits)
|
| 190 |
# st.write(logits.size())
|
|
|
|
| 232 |
percent_ans_topk_list = []
|
| 233 |
for i in range(k):
|
| 234 |
str_ans_topk_list.append(str_ans_topk[i].iloc[0])
|
| 235 |
+
percent_ans_topk_list.append(percent_ans_topk[i]*100)
|
| 236 |
|
| 237 |
# print(str_ans_topk_list)
|
| 238 |
|