Update app.py

app.py

@@ -28,7 +28,7 @@ def extract_organization_name(text):
 # Extract titles or identifiers for the ToS
 #tos_titles = [f"Document {i}" for i in range(len(dataset['train']))]
 
-tos_titles = [extract_organization_name(doc['plan_text']) for doc in dataset['train']]
+tos_titles = [extract_organization_name(doc['plain_text']) for doc in dataset['train']]
 
         
 # Set page to wide mode