Update app.py

app.py

@@ -34,8 +34,8 @@ def predict(message, max_new_tokens=4056, temperature=3.5, top_p=0.9, top_k=800)
         do_sample=True
     )
     response = tokenizer.decode(response_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
-    return response
-
+    return [("bot", response)]
+    
 DESCRIPTION = """
 # 👋🏻Welcome to 🙋🏻‍♂️Tonic's🧑🏻‍🚀YI-200K🚀"
 You can use this Space to test out the current model [Tonic/YI](https://huggingface.co/01-ai/Yi-34B)