Update app.py

app.py

@@ -61,7 +61,7 @@ def continue_writing(message, history, system_message, max_tokens, temperature,
     
     prompt = f"이전 대화를 간단히 요약하고 이어서 작성해주세요. 마지막 응답: {last_assistant_message[:100]}..."
     
-    async for response in respond(prompt, truncated_history[:-1], system_message, max_tokens, temperature, top_p, selected_model):
+    for response in respond(prompt, truncated_history[:-1], system_message, max_tokens, temperature, top_p, selected_model):
         yield response
 
 def stop_generation():
@@ -99,7 +99,9 @@ with gr.Blocks() as demo:
     # Event handlers
     msg.submit(respond, [msg, chatbot, system_message, max_tokens, temperature, top_p, model], [chatbot])
     send.click(respond, [msg, chatbot, system_message, max_tokens, temperature, top_p, model], [chatbot])
-    continue_btn.click(continue_writing, [msg, chatbot, system_message, max_tokens, temperature, top_p, model], [chatbot])
+    continue_btn.click(continue_writing, 
+        inputs=[msg, chatbot, system_message, max_tokens, temperature, top_p, model],
+        outputs=[chatbot])
     regenerate_btn.click(regenerate, [chatbot, system_message, max_tokens, temperature, top_p, model], [chatbot])
     stop.click(stop_generation, outputs=[msg])
     clear.click(lambda: None, outputs=[chatbot])