Update app.py

app.py

@@ -35,8 +35,8 @@ if st.button("判定", key="run"):
         pred_class, input_ids, attention_list = classify_ma(user_input)
         st.markdown(f"判定結果: **{ID2CAT[pred_class]}**")
         if pred_class == 1:
-            topic_dist, ll = infer_topic(user_input)
-            
+            topic_dist,new_word_list, ll = infer_topic(user_input)
+            print(new_word_list) #この一文を追加
             words_atten = get_word_attn(input_ids, attention_list)
             html_hext_result = html_hext(((word, attn) for word, attn in words_atten)) 
             st.markdown(html_hext_result, unsafe_allow_html=True)