Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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
|