Update app.py

app.py

@@ -180,7 +180,7 @@ def organize_faces_by_person(embeddings_by_frame, clusters, aligned_faces_folder
         shutil.copy(src, dst)
 
 def save_person_data_to_csv(embeddings_by_frame, emotions_by_frame, clusters, desired_fps, original_fps, output_folder, num_components):
-    emotions = ['angry', 'disgust', 'fear', 'happy', 'sad', 'neutral']
+    emotions = ['angry', 'disgust', 'fear', 'happy', 'sad', 'surprise', 'neutral']
     person_data = {}
 
     for (frame_num, embedding), (_, emotion_dict), cluster in zip(embeddings_by_frame.items(),