Gladiator commited on
Commit
6514554
1 Parent(s): b316c06

fix rogue score import

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -12,7 +12,7 @@ from utils import (
12
  read_text_from_file,
13
  )
14
 
15
- from rouge import rouge
16
 
17
  if __name__ == "__main__":
18
  # ---------------------------------
@@ -124,5 +124,6 @@ if __name__ == "__main__":
124
  st.info(summarized_text)
125
 
126
  st.subheader("Rogue Scores")
127
- score = rouge.get_scores(summarized_text, cleaned_txt, avg=True)
128
- st.write(score)
 
 
12
  read_text_from_file,
13
  )
14
 
15
+ from rouge import Rouge
16
 
17
  if __name__ == "__main__":
18
  # ---------------------------------
 
124
  st.info(summarized_text)
125
 
126
  st.subheader("Rogue Scores")
127
+ rouge_sc = Rouge()
128
+ score = rouge_sc.get_scores(summarized_text, cleaned_txt, avg=True)
129
+ st.code(score)