Update app.py

app.py

@@ -8,6 +8,8 @@ model_path = "cognitivecomputations/dolphin-2.8-mistral-7b-v02"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path)
 
+model.config.pad_token_id = model.config.eos_token_id
+
 @spaces.GPU
 def chat(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")