.

app.py

@@ -33,13 +33,13 @@ def predict_answer(question):
 demo = gr.Interface(
     fn=predict_answer,
     inputs=[
-        gr.Textbox(label='Question', value='A set of 7 spoons costs $21. If each spoon would be sold separately, how much would 5 spoons cost?'),
+        gr.Textbox(label='Question', value='Asumi\'s bookshelf has 120 books. She has 10 books on history, twice that many books on literature, and the rest are science books. How many science books does Asumi have?'),
     ],
     outputs=[
         gr.Textbox(label='Implicit CoT Prediction'),
     ],
-    title='Solving Grade School Math Problems with Implicit CoT',
-    description='This demo showcases Mistral-7B\'s ability to solve grade school math problems without producing intermediate steps, using our stepwise internalization method.',
+    title='Solving Grade School Math Problems without Intermediate Reasoning Steps',
+    description='This demo showcases Mistral-7B\'s ability to solve grade school math problems without producing intermediate steps, using our stepwise internalization approach linked below.',
     article="""
     - [Paper 1: Implicit Chain of Thought Reasoning via Knowledge Distillation](https://arxiv.org/pdf/2311.01460)
     - [Paper 2: From Explicit CoT to Implicit CoT: Learning to Internalize CoT Step by Step](https://arxiv.org/pdf/2405.14838)