Update app.py

app.py

@@ -126,11 +126,12 @@ def main():
                 rouge = Rouge()
                 scores = rouge.get_scores(st.session_state.summary, reference_summary)
     
-                # Display ROUGE scores
-                st.write("ROUGE Scores:")
-                st.write(f"ROUGE-1: {scores[0]['rouge-1']['f']:.4f}")
-                st.write(f"ROUGE-2: {scores[0]['rouge-2']['f']:.4f}")
-                st.write(f"ROUGE-L: {scores[0]['rouge-l']['f']:.4f}")
+                 # Display ROUGE scores in a box
+                with st.container():
+                    st.write("ROUGE Scores:")
+                    st.write(f"ROUGE-1: {scores[0]['rouge-1']['f']:.4f}", unsafe_allow_html=True)
+                    st.write(f"ROUGE-2: {scores[0]['rouge-2']['f']:.4f}", unsafe_allow_html=True)
+                    st.write(f"ROUGE-L: {scores[0]['rouge-l']['f']:.4f}", unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()