Update app.py

app.py

@@ -44,6 +44,7 @@ def generate(
     for response in stream:
         output_text += response.token.text
         yield output_text, history  # Yield the updated output_text and the updated history
+    yield output
 
 additional_inputs = [
     gr.Textbox(label="System Prompt", max_lines=1, interactive=True),