iSpr commited on
Commit
eacdce1
โ€ข
1 Parent(s): 4d4157a

Update app.py

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