Update app.py

app.py

@@ -1,5 +1,8 @@
-import torch
+
 import time
+import pandas as pd
+
+import torch
 from keybert import KeyBERT
 from sentence_transformers import SentenceTransformer
 from keyphrase_vectorizers import KeyphraseCountVectorizer
@@ -93,16 +96,18 @@ def extract_paraphrased_sentences(article):
 
 doc = st.text_area("Enter a custom document")
 
-t5_keywords_df, original_keywords_df, unique_keywords_df, total_end = extract_paraphrased_sentences(doc)
-
-# extract_paraphrased_article(input_list[0])
-st.text(f'T5 PARAPHRASING RUNTIME: {total_end}\n')
-
-st.subheader('\nOriginal Keywords Extracted:\n\n')
-st.dataframe(original_keywords_df)
-
-st.subheader('\nT5 Keywords Extracted:\n\n')
-st.dataframe(t5_keywords_df)
-
-st.subheader('\nT5 Unique New Keywords Extracted:\n\n')
-st.dataframe(unique_keywords_df)
+if doc:
+    t5_keywords_df, original_keywords_df, unique_keywords_df, total_end = extract_paraphrased_sentences(doc)
+    
+    # extract_paraphrased_article(input_list[0])
+    st.text(f'T5 PARAPHRASING RUNTIME: {total_end}\n')
+    
+    st.subheader('\nOriginal Keywords Extracted:\n\n')
+    st.dataframe(original_keywords_df)
+    
+    st.subheader('\nT5 Keywords Extracted:\n\n')
+    st.dataframe(t5_keywords_df)
+    
+    st.subheader('\nT5 Unique New Keywords Extracted:\n\n')
+    st.dataframe(unique_keywords_df)
+