.

README.md

@@ -1,5 +1,5 @@
 ---
-title: Gpt2 Multiplication Predictor
+title: Implicit CoT Grade School Math Solver 
 emoji: 📈
 colorFrom: green
 colorTo: green
@@ -9,7 +9,7 @@ app_file: app.py
 pinned: false
 license: mit
 models:
-- yuntian-deng/gpt2-implicit-cot-multiplication
+- yuntian-deng/implicit-cot-math-mistral7b
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -25,7 +25,7 @@ def predict_answer(question):
             do_sample=False)
     print (outputs)
 
-    prediction = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    prediction = tokenizer.decode(outputs[0, input_ids.shape[-1]:], skip_special_tokens=True)
 
     return prediction