IEIT-Yuan commited on
Commit
8119f32
Β·
verified Β·
1 Parent(s): 7c83f01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def predict(message, history):
59
  # break
60
  # yield partial_message
61
  inputs = tokenizer(message, return_tensors="pt")["input_ids"].to(device)
62
- outputs = model.generate(inputs, do_sample=False, max_length=100)
63
  print(tokenizer.decode(outputs[0]))
64
  return(tokenizer.decode(outputs[0]))
65
 
 
59
  # break
60
  # yield partial_message
61
  inputs = tokenizer(message, return_tensors="pt")["input_ids"].to(device)
62
+ outputs = model.generate(inputs, do_sample=False, max_length=500)
63
  print(tokenizer.decode(outputs[0]))
64
  return(tokenizer.decode(outputs[0]))
65