npc0 commited on
Commit
4fb5fa5
1 Parent(s): d901170

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -36,12 +36,14 @@ def sum_chain_l1(text, p_bar):
36
  for i in p_bar(range(len(text)//8000+1)):
37
  t = text[i*8000:i*8000+8196]
38
  if len(t) > 0:
 
39
  for answer in pipeline.chat(
40
  [chatglm_cpp.ChatMessage(
41
  role="user",
42
  content=prompt+t)],
43
  stream=True):
44
- yield f"{'='*8} {i+1}/{len(text)//8000+1} {'='*8}\n{answer.content}"
 
45
  docs.append(answer)
46
  yield docs
47
 
@@ -50,12 +52,14 @@ def sum_chain_l2_deprecated(docs, p_bar):
50
  i = 0
51
  for doc in p_bar(docs[1:]):
52
  i += 1
 
53
  for answer in pipeline.chat(
54
  [chatglm_cpp.ChatMessage(
55
  role="user",
56
  content=prompt+"\n"+hist+"\n"+doc)],
57
  stream=True):
58
- yield f"{'='*8} {i}/{len(docs)} {'='*8}\n{answer.content}"
 
59
  hist = answer
60
  yield hist
61
 
 
36
  for i in p_bar(range(len(text)//8000+1)):
37
  t = text[i*8000:i*8000+8196]
38
  if len(t) > 0:
39
+ a = ''
40
  for answer in pipeline.chat(
41
  [chatglm_cpp.ChatMessage(
42
  role="user",
43
  content=prompt+t)],
44
  stream=True):
45
+ a += answer.content
46
+ yield f"{'='*8} {i+1}/{len(text)//8000+1} {'='*8}\n{a}"
47
  docs.append(answer)
48
  yield docs
49
 
 
52
  i = 0
53
  for doc in p_bar(docs[1:]):
54
  i += 1
55
+ a = ''
56
  for answer in pipeline.chat(
57
  [chatglm_cpp.ChatMessage(
58
  role="user",
59
  content=prompt+"\n"+hist+"\n"+doc)],
60
  stream=True):
61
+ a += answer.content
62
+ yield f"{'='*8} {i}/{len(docs)} {'='*8}\n{a}"
63
  hist = answer
64
  yield hist
65