Update app.py

app.py

@@ -36,6 +36,7 @@ if st.button("判定", key="run"):
         st.markdown(f"判定結果: **{ID2CAT[pred_class]}**")
         if pred_class == 1:
             topic_dist, ll = infer_topic(user_input)
+            st.markdown(topic_dist)
             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)