Update app.py

app.py

@@ -80,6 +80,7 @@ class ChatBot:
                 max_new_tokens=2048,
                 temperature=0.6,
                 do_sample=True,
+                return_dict_in_generate=True,
                 pad_token_id=self.tokenizer.eos_token_id,
                 eos_token_id=[self.tokenizer.eos_token_id] + stop_token_ids,
             )