Update app.py

app.py

@@ -37,10 +37,11 @@ def sum_chain_l1(text, p_bar):
         t = text[i*8000:i*8000+8196]
         if len(t) > 0:
             for answer in pipeline.chat(
-                            chatglm_cpp.ChatMessage(role="user",
-                                                    content=prompt+t),
+                            [chatglm_cpp.ChatMessage(
+                                role="user",
+                                content=prompt+t)],
                             stream=True):
-                yield f"{'='*8} {i+1}/{len(text)//8000+1} {'='*8}\n{answer}"
+                yield f"{'='*8} {i+1}/{len(text)//8000+1} {'='*8}\n{answer.content}"
             docs.append(answer)
     yield docs
 
@@ -50,10 +51,11 @@ def sum_chain_l2_deprecated(docs, p_bar):
     for doc in p_bar(docs[1:]):
         i += 1
         for answer in pipeline.chat(
-                            chatglm_cpp.ChatMessage(role="user",
-                                                    content=prompt+"\n"+hist+"\n"+doc),
+                            [chatglm_cpp.ChatMessage(
+                                role="user",
+                                content=prompt+"\n"+hist+"\n"+doc)],
                             stream=True):
-            yield f"{'='*8} {i}/{len(docs)} {'='*8}\n{answer}"
+            yield f"{'='*8} {i}/{len(docs)} {'='*8}\n{answer.content}"
         hist = answer
     yield hist