Update app.py

app.py

@@ -127,7 +127,7 @@ class MistralChatBot:
             batch_prompts = [editorial_text]
 
         out = token_classifier(batch_prompts)
-        out = transform_chunks(out)
+        out = transform_chunks(out[0])
         print(out)
         generated_text = '<h2 style="text-align:center">Réponse</h3>\n<div class="generation">' + out + "</div>"
         return generated_text