Update README.md

README.md

@@ -103,18 +103,14 @@ class ChatBot:
         user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors="pt")
 
         # Concatenate the user input with chat history
-        if self.history:
+        if self.history.numel() > 0:
             chat_history_ids = torch.cat([self.history, user_input_ids], dim=-1)
         else:
             chat_history_ids = user_input_ids
 
         # Generate a response using the PEFT model
-        # response = peft_model.generate(chat_history_ids, max_length=512, pad_token_id=tokenizer.eos_token_id)
-        # response = peft_model.generate(chat_history_ids)  
         response = peft_model.generate(input_ids=chat_history_ids, max_length=512, pad_token_id=tokenizer.eos_token_id)
 
-
-
         # Update chat history
         self.history = chat_history_ids