.

app.py

@@ -21,9 +21,8 @@ def predict_answer(question):
     outputs = implicit_cot_model.generate(input_ids=input_ids,
             max_new_tokens=MAX_RESULT_TOKENS,
             do_sample=False)
-    generated_ids = outputs.sequences[0]
 
-    prediction = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
+    prediction = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
     return prediction