Update app.py

app.py

@@ -8,6 +8,10 @@ tokenizer_name = "HuggingFaceTB/SmolLM-135M"
 def load_model_and_tokenizer():
     tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)
+    # Ensure tokenizer.pad_token is set to tokenizer.eos_token
+    if tokenizer.pad_token is None:
+        tokenizer.pad_token = tokenizer.eos_token
+        tokenizer.pad_token_id = tokenizer.eos_token_id
     return tokenizer, model
 
 tokenizer, model = load_model_and_tokenizer()