Update app.py

app.py

@@ -30,7 +30,7 @@ def main():
     st.title("Terms of Service Summarizer")
 
     # Layout: 3 columns
-    col1, col2, col3 = st.columns([1, 2, 3], gap="large")
+    col1, col2, col3 = st.columns([1, 3, 2], gap="large")
 
     # Left column: Radio buttons for summarizer choice
     with col1:
@@ -119,7 +119,7 @@ def main():
             # Check if a reference summary is available
             if tos_selection_index is not None and 'summary' in dataset['train'][tos_selection_index]:
                 # Fetch the reference summary
-                reference_summary = data['train'][tos_selection_index]['summary']
+                reference_summary = dataset['train'][tos_selection_index]['summary']
     
                 # Calculate ROUGE scores
                 rouge = Rouge()