Spaces:
Running
Running
acecalisto3
commited on
Commit
•
ae4ec08
1
Parent(s):
8091f1f
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
|