Update app.py

app.py

@@ -128,6 +128,10 @@ class Chatbot:
         """
         self.tokenizer = AutoTokenizer.from_pretrained(model_name)
         self.model = AutoModelForCausalLM.from_pretrained(model_name)
+        
+        # Set pad_token to eos_token if not already defined
+        if self.tokenizer.pad_token is None:
+            self.tokenizer.pad_token = self.tokenizer.eos_token
 
     def generate_response(self, prompt, max_length=100):
         """
@@ -142,7 +146,7 @@ class Chatbot:
             attention_mask=inputs.attention_mask,  # Pass the attention mask
             max_length=max_length,
             num_return_sequences=1,
-            pad_token_id=self.tokenizer.eos_token_id,  # Set pad_token_id to eos_token_id
+            pad_token_id=self.tokenizer.pad_token_id,  # Use the defined pad_token_id
         )
         
         # Decode the generated response