Update app.py

app.py

@@ -54,7 +54,7 @@ def sum_chain_l1(text, p_bar):
                             temperature=temperature):
                 yield f"{'='*8} {i+1}/{len(text)//2000+1} {'='*8}\n{answer}"
             docs.append(answer)
-    return docs
+    yield docs
 
 def sum_chain_l2_deprecated(docs, p_bar):
     hist = docs[0]
@@ -68,7 +68,7 @@ def sum_chain_l2_deprecated(docs, p_bar):
                     temperature=temperature):
             yield f"{'='*8} {i}/{len(docs)} {'='*8}\n{answer}"
         hist = answer
-    return hist
+    yield hist
 
 import gradio as gr
 
@@ -76,12 +76,13 @@ def greet(text, progress=gr.Progress()):
     progress(0, desc="Reading")
     docs = []
     for doc in sum_chain_l1(text, progress.tqdm):
-        yield '# drafting summary', doc
+        if isinstance(doc, str):
+            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
+        if isinstance(ans, str):
+            yield '# refining summary', ans
     return '# final result', ans
 
 gr.Markdown("# Text Summarization\nStage 1 = draft -> Stage 2 = refine -> final result")