Update app.py

app.py

@@ -22,7 +22,7 @@ emotion_labels = ["admiration", "amusement", "anger", "annoyance", "approval",
                   "neutral"]
 
 # Function to classify emotions in batches
-def classify_emotions_in_batches(texts, batch_size=1024):
+def classify_emotions_in_batches(texts, batch_size=32):
     results = []
     start_time = time.time()
     for i in range(0, len(texts), batch_size):
@@ -53,7 +53,7 @@ if st.button("Run Inference"):
     # Apply emotion classification to the email content
     with st.spinner('Running inference...'):
         email_texts = enron_data['body'].tolist()
-        enron_data['emotion'] = classify_emotions_in_batches(email_texts, batch_size=1024)
+        enron_data['emotion'] = classify_emotions_in_batches(email_texts, batch_size=32)
 
     # Save the results to a CSV file
     enron_data.to_csv("enron_emails_with_emotions.csv", index=False)