diff --git a/api/test/test.py b/api/test/test.py index 690622c..baba984 100644 --- a/api/test/test.py +++ b/api/test/test.py @@ -6,6 +6,7 @@ while True: break else: packet = { + 'bot_name': 'cartman', 'message': user_input, 'max_new_tokens': 20, 'num_beams': 2,