Update app.py

app.py

@@ -49,7 +49,7 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
         **generate_kwargs,
     )
 
-    output = ""
+    output = prompt
     for response in stream:
         output += response.token.text
         yield output