diff --git a/web_demo.py b/web_demo.py index ff7fc700..315978e2 100644 --- a/web_demo.py +++ b/web_demo.py @@ -9,7 +9,9 @@ MAX_BOXES = MAX_TURNS * 2 -def predict(input, history=[]): +def predict(input, history=None): + if history is None: + history = [] response, history = model.chat(tokenizer, input, history) updates = [] for query, response in history: