Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline, AutoTokenizer
 import base64
 
 # Load the EasyTerms/legalSummerizerET model from Hugging Face
-summarizer = pipeline("summarization", model="EasyTerms/legalSummerizerET", num_labels=1)
+summarizer = pipeline("summarization", model="EasyTerms/legalSummerizerET")
 
 # Increase the maximum token limit
 tokenizer = AutoTokenizer.from_pretrained("EasyTerms/legalSummerizerET")
@@ -22,6 +22,9 @@ def handle_feedback(feedback_data, feedback_file):
     # Save the dataframe to the feedback CSV file
     feedback_df.to_csv(feedback_file, mode='a', index=False, header=not st.session_state.feedback_csv_exists)
 
+    # Display a feedback collected message
+    st.success("Feedback collected successfully!")
+
     # Display a download button for the user
     download_button = st.button("Download Feedback CSV")
 
@@ -59,11 +62,11 @@ def main():
 
         feedback_data.append((contract_text, summary, chosen, rejected))
 
-    # Handle feedback data
-    if feedback_data:
-        feedback_file = 'feedback.csv'
-        st.session_state.feedback_csv_exists = True
-        handle_feedback(feedback_data, feedback_file)
+        # Handle feedback data
+        if feedback_data:
+            feedback_file = 'feedback.csv'
+            st.session_state.feedback_csv_exists = True
+            handle_feedback(feedback_data, feedback_file)
 
 # Initialize feedback data
 feedback_data = []