Update space

app.py

@@ -104,10 +104,10 @@ class ChatLLM:
             repeat_penalty=repeat_penalty,
             stop=TOKEN_STOP,
         ):
-            answer = output["choices"][0]["text"]
+            answer = output
             history[-1][1] += answer # here we append the answer to the last message in the history
             # stream the response
-            yield history, answer
+            yield history, history
 
 
 def user(message, history):