Update app.py

app.py

@@ -10,6 +10,13 @@ def predict(input, history=[]):
   bot_input_ids = torch.cat([torch.LongTensor(history), new_user_input_ids], dim=-1)
   history = model.generate(bot_input_ids, max_length=500, pad_token_id=tokenizer.eos_token_id).tolist()
   response = tokenizer.decode(history[0]).replace("<|endoftext|>", "\n")
+  
+  html = "<div class='chatbot'>"
+    for m, msg in enumerate(response):
+      cls = "user" if m%2 == 0 else "bot"
+      html += "<div class='msg {}'> {}</div>".format(cls, msg)
+    html += "</div>"
+  
   return response, history
 
 css = """