Tonic commited on
Commit
ec3875c
•
1 Parent(s): c2970ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ class ChatBot:
69
  chat_history_ids = user_input_ids
70
 
71
  # Generate a response using the PEFT model
72
- response = model.generate(input_ids=chat_history_ids, max_length=512, pad_token_id=tokenizer.eos_token_id)
73
 
74
  # Update chat history
75
  self.history = chat_history_ids
 
69
  chat_history_ids = user_input_ids
70
 
71
  # Generate a response using the PEFT model
72
+ response = model.generate(input_ids=chat_history_ids, max_length=200, pad_token_id=tokenizer.eos_token_id)
73
 
74
  # Update chat history
75
  self.history = chat_history_ids