Update app.py

app.py

@@ -97,13 +97,8 @@ def transform_chunks(marianne_segmentation):
     
     # Add entity_group as a header to each word
     marianne_segmentation['word'] = '### ' + marianne_segmentation['entity_group'] + ' ###\n' + marianne_segmentation['word']
-    
-    # Group by text_id, identifier, and date, then concatenate words
-    marianne_segmentation = marianne_segmentation.agg({
-        'word': lambda x: '\n\n'.join(x.dropna())
-    }).reset_index()
 
-    final_text = marianne_segmentation['word'].tolist()[0]
+    final_text = '\n\n'.join(marianne_segmentation['word'].tolist())
 
     return final_text