da03 commited on
Commit
15cb83f
·
1 Parent(s): d87049c
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -21,9 +21,8 @@ def predict_answer(question):
21
  outputs = implicit_cot_model.generate(input_ids=input_ids,
22
  max_new_tokens=MAX_RESULT_TOKENS,
23
  do_sample=False)
24
- generated_ids = outputs.sequences[0]
25
 
26
- prediction = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
27
 
28
  return prediction
29
 
 
21
  outputs = implicit_cot_model.generate(input_ids=input_ids,
22
  max_new_tokens=MAX_RESULT_TOKENS,
23
  do_sample=False)
 
24
 
25
+ prediction = tokenizer.decode(outputs[0], skip_special_tokens=True)
26
 
27
  return prediction
28