Tonic commited on
Commit
40b7c4e
1 Parent(s): ab587f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ class ChatBot:
80
  chat_history_ids = user_input_ids
81
 
82
  # Generate a response using the PEFT model
83
- response = peft_model.generate(input_ids=chat_history_ids, max_length=400, pad_token_id=tokenizer.eos_token_id)
84
 
85
  # Update chat history
86
  self.history = chat_history_ids
 
80
  chat_history_ids = user_input_ids
81
 
82
  # Generate a response using the PEFT model
83
+ response = peft_model.generate(input_ids=chat_history_ids, max_length=1200, pad_token_id=tokenizer.eos_token_id)
84
 
85
  # Update chat history
86
  self.history = chat_history_ids