acecalisto3 commited on
Commit
ae4ec08
1 Parent(s): 8091f1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, max_length=50, num_return_sequences=1, do_sample=True)[0]['generated_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