Update app.py

app.py

@@ -7,11 +7,13 @@ title = "EZChat"
 description = "A State-of-the-Art Large-scale Pretrained Response generation model (DialoGPT-medium)"
 examples = [["How are you?"]]
 
-
+# Set the padding token to be used and initialize the model
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium", padding_side='left')
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
+tokenizer.add_special_tokens({'pad_token': '[EOS]'})
+tokenizer.pad_token = tokenizer.eos_token
 
-
+#predict
 def predict(input, history=[]):
     # tokenize the new input sentence
     new_user_input_ids = tokenizer.encode(