umang018 commited on
Commit
b521c82
·
verified ·
1 Parent(s): 313b807

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ if st.button("Run Inference"):
53
  # Apply emotion classification to the email content
54
  with st.spinner('Running inference...'):
55
  email_texts = enron_data['body'].tolist()
56
- enron_data['emotion'] = classify_emotions_in_batches(email_texts, batch_size=32)
57
 
58
  # Save the results to a CSV file
59
  enron_data.to_csv("enron_emails_with_emotions.csv", index=False)
 
53
  # Apply emotion classification to the email content
54
  with st.spinner('Running inference...'):
55
  email_texts = enron_data['body'].tolist()
56
+ enron_data['emotion'] = classify_emotions_in_batches(email_texts, batch_size=1024)
57
 
58
  # Save the results to a CSV file
59
  enron_data.to_csv("enron_emails_with_emotions.csv", index=False)