Update app.py

app.py

@@ -110,13 +110,14 @@ def transform_chunks(marianne_segmentation):
 
     html_output = []
     for _, row in marianne_segmentation.iterrows():
-        entity_group = "<" + capitalize(row['entity_group']) + ">"
+        entity_group = row['entity_group']
+        result_entity = "<" + entity_group.capitalize() + ">"
         word = row['word']
         
         if entity_group == 'title':
-            html_output.append(f'<div class="manuscript"><div class="annotation">{entity_group}</div><div class="content title-content"><h2>{word}</h2></div></div>')
+            html_output.append(f'<div class="manuscript"><div class="annotation">{result_entity}</div><div class="content title-content"><h2>{word}</h2></div></div>')
         else:
-            html_output.append(f'<div class="manuscript"><div class="annotation">{entity_group}</div><div class="content">{word}</div></div>')
+            html_output.append(f'<div class="manuscript"><div class="annotation">{result_entity}</div><div class="content">{word}</div></div>')
 
     final_html = '\n'.join(html_output)
     return final_html
@@ -143,7 +144,7 @@ class MistralChatBot:
             classified_list.append(df)
     
         classified_list = pd.concat(classified_list)
-        out = transform_chunks(pd.concat(classified_list))
+        out = transform_chunks(classified_list)
         generated_text = f'{css}<h2 style="text-align:center">Réponse</h2>\n<div class="generation">{out}</div>'
         return generated_text