iSpr commited on
Commit
c606fbe
·
1 Parent(s): 3543f19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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