Update app.py

app.py

@@ -12,7 +12,7 @@ from keyphrase_extraction import extract_sentences_with_obligations
 dataset = load_dataset("EE21/ToS-Summaries")
 
 # Extract titles or identifiers for the ToS
-tos_titles = ["None"] + [f"Document {i}" for i in range(len(dataset['train']))]
+tos_titles = [f"Document {i}" for i in range(len(dataset['train']))]
 
 # Set page to wide mode
 st.set_page_config(layout="wide")
@@ -64,8 +64,7 @@ def main():
                 st.write("PDF uploaded successfully.")
             elif user_input:
                 file_content = user_input
-            elif tos_selection_index != "None":
-                tos_selection_index = tos_titles.index(tos_titles) - 1  # Adjust for the added 'None' option
+            elif tos_selection_index is not None:
                 file_content = dataset['train'][tos_selection_index]['plain_text']
             else:
                 st.warning("Please upload a PDF, enter some text, or select a document to summarize.")