Update app.py

app.py

@@ -52,19 +52,21 @@ def sum_chain_l1(text, p_bar):
                             top_k=top_k,
                             top_p=top_p,
                             temperature=temperature):
-                yield f"{'='*8} {i}/{len(text)//2000+1} {'='*8}\n{answer}"
+                yield f"{'='*8} {i+1}/{len(text)//2000+1} {'='*8}\n{answer}"
             docs.append(answer)
     return docs
 
 def sum_chain_l2_deprecated(docs, p_bar):
     hist = docs[0]
+    i = 0
     for doc in p_bar(docs[1:]):
+        i += 1
         for answer in model.generate_iterate(prompt+"\n"+hist+"\n"+doc,
                     max_generated_tokens=max_tokens,
                     top_k=top_k,
                     top_p=top_p,
                     temperature=temperature):
-            yield answer
+            yield f"{'='*8} {i}/{len(docs)} {'='*8}\n{answer}"
         hist = answer
     return hist
 
@@ -76,6 +78,7 @@ def greet(text, progress=gr.Progress()):
     for doc in sum_chain_l1(text, progress.tqdm):
         yield '# drafting summary', doc
     # yield 'stage 1 finished', '\n===== summarized parts =====\n'.join(doc)
+    print('stage 1 finished\n'+'\n===== summarized parts =====\n'.join(doc))
     progress(0, desc="Refinement")
     for ans in sum_chain_l2_deprecated(doc, progress.tqdm):
         yield '# refining summary', ans