Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -126,11 +126,12 @@ def main():
|
|
126 |
rouge = Rouge()
|
127 |
scores = rouge.get_scores(st.session_state.summary, reference_summary)
|
128 |
|
129 |
-
|
130 |
-
st.
|
131 |
-
|
132 |
-
|
133 |
-
|
|
|
134 |
|
135 |
if __name__ == "__main__":
|
136 |
main()
|
|
|
126 |
rouge = Rouge()
|
127 |
scores = rouge.get_scores(st.session_state.summary, reference_summary)
|
128 |
|
129 |
+
# Display ROUGE scores in a box
|
130 |
+
with st.container():
|
131 |
+
st.write("ROUGE Scores:")
|
132 |
+
st.write(f"ROUGE-1: {scores[0]['rouge-1']['f']:.4f}", unsafe_allow_html=True)
|
133 |
+
st.write(f"ROUGE-2: {scores[0]['rouge-2']['f']:.4f}", unsafe_allow_html=True)
|
134 |
+
st.write(f"ROUGE-L: {scores[0]['rouge-l']['f']:.4f}", unsafe_allow_html=True)
|
135 |
|
136 |
if __name__ == "__main__":
|
137 |
main()
|