Update app.py

app.py

@@ -34,18 +34,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