.

app.py

@@ -61,13 +61,13 @@ demo = gr.Interface(
         gr.Textbox(label='Second Number (up to 12 digits)', value='67890'),
     ],
     outputs=[
-        gr.Textbox(label='Raw Input to GPT-2 (reversed digits and added spaces)'),
-        gr.Textbox(label='Raw Output from GPT-2 (reversed digits and with spaces)'),
+        gr.Textbox(label='Raw Input to GPT2 (reversed digits and added spaces)'),
+        gr.Textbox(label='Raw Output from GPT2 (reversed digits and with spaces)'),
         gr.Textbox(label='Predicted Product'),
         gr.HTML(label='Result Message')
     ],
-    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.',
+    title='GPT2 Direct Multiplication Calculator (Without Using Chain-of-Thought)',
+    description='This demo uses GPT2 to directly predict the product of two numbers without using any intermediate reasoning steps. The GPT2 model has been fine-tuned to internalize chain-of-thought reasoning within its hidden states, following our stepwise internalization approach detailed in the paper linked at the bottom of this page.',
     article="""
     - [Paper: From Explicit CoT to Implicit CoT: Learning to Internalize CoT Step by Step](https://arxiv.org/pdf/2405.14838)
     - [Code Repository](https://github.com/da03/Internalize_CoT_Step_by_Step)