mrcuddle commited on
Commit
4c72205
·
verified ·
1 Parent(s): 36b0abf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def chat(message, history):
15
 
16
  # Generate a response
17
  with torch.no_grad():
18
- outputs = model.generate(inputs.input_ids, max_length=50, num_return_sequences=1)
19
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
20
 
21
  # Extract only the new response part
 
15
 
16
  # Generate a response
17
  with torch.no_grad():
18
+ outputs = model.generate(inputs.input_ids, max_length=300, num_return_sequences=1)
19
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
20
 
21
  # Extract only the new response part