BeveledCube commited on
Commit
f939f23
·
verified ·
1 Parent(s): 7a4525a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -41,7 +41,7 @@ def read_root(data: req):
41
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
42
 
43
  # Generate output using the model
44
- output_ids = model.generate(input_ids, max_length=length, num_beams=5, no_repeat_ngram_size=2)
45
  generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
46
 
47
  answer_data = { "answer": generated_text }
 
41
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
42
 
43
  # Generate output using the model
44
+ output_ids = model.generate(input_ids, max_length=data.length, num_beams=5, no_repeat_ngram_size=2)
45
  generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
46
 
47
  answer_data = { "answer": generated_text }