Update app.py

app.py

@@ -32,8 +32,6 @@ if st.button("判定", key="run"):
     if not user_input:
         st.warning("入力が空です。何か入力してください。")
     else:
-        st.markdown(attention_text)
-        st.divider()
         pred_class, input_ids, attention_list = classify_ma(user_input)
         st.markdown(f"判定結果: **{ID2CAT[pred_class]}**")
         if pred_class == 1:
@@ -46,6 +44,10 @@ if st.button("判定", key="run"):
             data = topic_dist.reshape(-1, 1)
             st.plotly_chart(heatmap(data), use_container_width=True)
 
+        st.divider()
+        st.markdown(attention_text)
+
+
 
 st.divider()
 st.markdown(provide_by)