summ

app.py

@@ -61,15 +61,39 @@ def main():
 
             with c2:
                 with st.expander("TextRank Summary"):
-                    text_summary=sumy_text_summarizer(raw_text)
+                    try:
+                        text_summary=sumy_text_summarizer(raw_text)
+                        document_len={"Original":len(raw_text),
+                                  "Summary":len(summary)
+                                   }
+                        st.write(document_len)
+                        st.write(text_summary)
+
+                        st.info("Rouge Score")
+                        score=evaluate_summary(text_summary,raw_text)
+                        st.write(score.T)
+                        st.subheader(" ")
+                        score['metrics']=score.index
+                        c=at.Chart(score).mark_bar().encode(
+                            x='metrics',y='rouge-1'
+                        )
+                        st.altair_chart(c)
+                        
+                    except:
+                          st.warning('Insufficient data')
+                        
+
+            st.subheader("Bart Sumary")
+            with st.expander("Bart Summary"):
+                try:
+                    bart_summ = bart_summary(raw_text)
                     document_len={"Original":len(raw_text),
                                   "Summary":len(summary)
                                    }
                     st.write(document_len)
-                    st.write(text_summary)
-
+                    st.write(bart_summ)
                     st.info("Rouge Score")
-                    score=evaluate_summary(text_summary,raw_text)
+                    score=evaluate_summary(bart_summ,raw_text)
                     st.write(score.T)
                     st.subheader(" ")
                     score['metrics']=score.index
@@ -77,24 +101,8 @@ def main():
                         x='metrics',y='rouge-1'
                     )
                     st.altair_chart(c)
-
-            st.subheader("Bart Sumary")
-            with st.expander("Bart Summary"):
-                bart_summ = bart_summary(raw_text)
-                document_len={"Original":len(raw_text),
-                                  "Summary":len(summary)
-                                   }
-                st.write(document_len)
-                st.write(bart_summ)
-                st.info("Rouge Score")
-                score=evaluate_summary(bart_summ,raw_text)
-                st.write(score.T)
-                st.subheader(" ")
-                score['metrics']=score.index
-                c=at.Chart(score).mark_bar().encode(
-                        x='metrics',y='rouge-1'
-                    )
-                st.altair_chart(c)
+                except:
+                      st.warning('Insufficient data')
 
             st.subheader("T5 Sumarization")
             with st.expander("T5 Summary"):