Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ def chat_interface(input_text):
|
|
| 27 |
return f'Error loading model: {e}'
|
| 28 |
|
| 29 |
# Generate chatbot response
|
| 30 |
-
response = generator(input_text,
|
| 31 |
return response
|
| 32 |
|
| 33 |
# 2. Terminal
|
|
|
|
| 27 |
return f'Error loading model: {e}'
|
| 28 |
|
| 29 |
# Generate chatbot response
|
| 30 |
+
response = generator(input_text, max_new_tokens=50, num_return_sequences=1, do_sample=True)[0]['generated_text']
|
| 31 |
return response
|
| 32 |
|
| 33 |
# 2. Terminal
|