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