Update app.py
Browse files
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=
|
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)
|