.

app.py

@@ -58,8 +58,8 @@ demo = gr.Interface(
         gr.Textbox(label='Predicted Product'),
         gr.HTML(label='Result Message')
     ],
-    title='GPT-2 Multiplication Calculator',
-    description='This demo uses GPT-2 to directly predict the product of two numbers without using any intermediate steps.',
+    title='GPT-2 Direct Multiplication Calculator (Without Using Chain-of-Thought)',
+    description='This demo uses GPT-2 to directly predict the product of two numbers without using any intermediate steps. The GPT-2 is finetuned to internalize chain-of-thought reasoning in its hidden states, using our stepwise internalization approach detailed in the paper below.',
     article="""
     ### Additional Resources
     - [Paper: From Explicit CoT to Implicit CoT: Learning to Internalize CoT Step by Step](https://arxiv.org/pdf/2405.14838)