Update app.py

app.py

@@ -137,7 +137,12 @@ class MistralChatBot:
             batch_prompts = [editorial_text]
     
         out = token_classifier(batch_prompts)
-        out = transform_chunks(pd.concat(out))
+        classified_list = []
+        for classification in out:
+            df = pd.DataFrame(classification)
+    
+        classified_list = pd.concat(classified_list)
+        out = transform_chunks(pd.concat(classified_list))
         generated_text = f'{css}<h2 style="text-align:center">Réponse</h2>\n<div class="generation">{out}</div>'
         return generated_text