Spaces:

vibey
/

article-summariser-for-final-project

Runtime error

App Files Files Community

vibey commited on Aug 17, 2022

Commit

e7c886a

1 Parent(s): d4ce5ca

Update app.py

Browse files

Files changed (1) hide show

app.py +6 -6

app.py CHANGED Viewed

@@ -65,8 +65,8 @@ def main():
                 with st.spinner(
                     text="Loading Bart Model and Extracting summary. This might take a few seconds depending on the length of your text..."):
                     summarizer_model = bart()
-                    global summarized_text = summarizer_model(text_to_summarize, max_length=max_length ,min_length=min_length)
-                    global summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
             elif model_type == "T5":
                 text_to_summarize = clean_text
@@ -74,8 +74,8 @@ def main():
                 with st.spinner(
                     text="Loading T5 Model and Extracting summary. This might take a few seconds depending on the length of your text..."):
                     summarizer_model = t5()
-                    global summarized_text = summarizer_model(text_to_summarize, max_length=max_length, min_length=min_length)
-                    global summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
             elif model_type == "ssr":
                 text_to_summarize = clean_text
@@ -83,8 +83,8 @@ def main():
                 with st.spinner(
                     text="Loading SSR Model and Extracting summary. This might take a few seconds depending on the length of your text..."):
                     summarizer_model = ssr()
-                    global summarized_text = summarizer_model(text_to_summarize, max_length=max_length, min_length=min_length)
-                    global summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
             res_col1 ,res_col2 = st.columns(2)
             with res_col1:

                 with st.spinner(
                     text="Loading Bart Model and Extracting summary. This might take a few seconds depending on the length of your text..."):
                     summarizer_model = bart()
+                    summarized_text = summarizer_model(text_to_summarize, max_length=max_length ,min_length=min_length)
+                    summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
             elif model_type == "T5":
                 text_to_summarize = clean_text
                 with st.spinner(
                     text="Loading T5 Model and Extracting summary. This might take a few seconds depending on the length of your text..."):
                     summarizer_model = t5()
+                    summarized_text = summarizer_model(text_to_summarize, max_length=max_length, min_length=min_length)
+                    summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
             elif model_type == "ssr":
                 text_to_summarize = clean_text
                 with st.spinner(
                     text="Loading SSR Model and Extracting summary. This might take a few seconds depending on the length of your text..."):
                     summarizer_model = ssr()
+                    summarized_text = summarizer_model(text_to_summarize, max_length=max_length, min_length=min_length)
+                    summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
             res_col1 ,res_col2 = st.columns(2)
             with res_col1: