Update app.py

app.py

@@ -35,7 +35,7 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=904
     output = ""
     for response in stream:
         output += response.token.text
-    return output
+        yield output
 
 additional_inputs = [
     gr.Textbox(label="System Prompt", max_lines=1, interactive=True),