Update app.py

app.py

@@ -73,8 +73,8 @@ def generate(
 
     for response in stream:
         output += response.token.text
-        yield output
-    return output
+        yield [prompt,output]
+    return [prompt,output]
 
 
 additional_inputs=[