Update app.py

app.py

@@ -40,7 +40,11 @@ class ChatBot:
             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, 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