Spaces:
Runtime error
Runtime error
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 |
|