Tonic commited on
Commit
023b8d0
Β·
verified Β·
1 Parent(s): bf2632f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def generate_text(usertitle, content, max_length, temperature):
25
  # }]
26
  input_text = [{Title: {usertitle}, Content: {content}}]
27
  inputs = tokenizer.apply_chat_template(input_text, return_tensors='pt').cuda()
28
- genrated_text = tokenizer.decode(model.generate(inputs, max_new_tokens=max_length, temperature=temperature, do_sample=True)[0]).split(tokenizer.eos_token)[0])
29
 
30
  return generated_text
31
 
 
25
  # }]
26
  input_text = [{Title: {usertitle}, Content: {content}}]
27
  inputs = tokenizer.apply_chat_template(input_text, return_tensors='pt').cuda()
28
+ genrated_text = tokenizer.decode(model.generate(inputs, max_new_tokens=max_length, temperature=temperature, do_sample=True)[0]).split(tokenizer.eos_token)[0]
29
 
30
  return generated_text
31