.

app.py

@@ -137,12 +137,12 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.HighlightedText(label='Ground Truth Product', combine_adjacent=False, show_legend=False, color_map=color_map),
-        gr.HighlightedText(label='Implicit CoT Predicted Product', combine_adjacent=False, show_legend=False, color_map=color_map, show_inline_category=False),
-        gr.HighlightedText(label='No CoT Predicted Product', combine_adjacent=False, show_legend=False, color_map=color_map, show_inline_category=False),
-        gr.HighlightedText(label='Explicit CoT Predicted Intermediate Steps & Product', combine_adjacent=False, show_legend=False, color_map=color_map, show_inline_category=False),
+        gr.HighlightedText(label='Implicit CoT Prediction', combine_adjacent=False, show_legend=False, color_map=color_map, show_inline_category=False),
+        gr.HighlightedText(label='No CoT Prediction', combine_adjacent=False, show_legend=False, color_map=color_map, show_inline_category=False),
+        gr.HighlightedText(label='Explicit CoT Steps & Prediction', combine_adjacent=False, show_legend=False, color_map=color_map, show_inline_category=False),
     ],
-    title='Can GPT2 Predict Multiplication of Two Numbers Without Intermediate Steps?',
-    description='This demo demonstrates GPT2\'s ability to directly predict the product of two large numbers without intermediate reasoning steps. The GPT2 has been finetuned to internalize chain-of-thought (CoT) reasoning within its hidden states through our stepwise internalization approach. The results demonstrate the effectiveness of implicit CoT (our approach, accurate and fast), compared to no CoT (fast but inaccurate) and explicit CoT (accurate but slow).',
+    title='Predicting Multiplication with GPT-2: Implicit vs. Explicit CoT',
+    description='This demo showcases GPT-2\'s ability to directly predict the product of two large numbers without intermediate steps, using our stepwise internalization method. Compare the performance of implicit CoT (our method), no CoT, and explicit CoT. Implicit CoT offers accuracy and speed, while explicit CoT provides detailed reasoning but is slower.',
     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)