Update app.py

app.py

@@ -34,9 +34,9 @@ if st.button("判定", key="run"):
     else:
         pred_class, input_ids, attention_list = classify_ma(user_input)
         st.markdown(f"判定結果: **{ID2CAT[pred_class]}**")
+        st.markdown(new_word_list) #この一文を追加
         if pred_class == 1:
             topic_dist,new_word_list, ll = infer_topic(user_input)
-            st.markdown(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)