.

app.py

@@ -119,13 +119,14 @@ def predict_product(num1, num2):
 
         yield ground_truth_digits_digits, predicted_results_implicit, predicted_results_nocot, predicted_results_explicit_cot
 
+color_map = {"correct": "green", "wrong": "red"}
+
 demo = gr.Interface(
     fn=predict_product,
     inputs=[
         gr.Textbox(label='First Number (up to 12 digits)', value='123456789'),
         gr.Textbox(label='Second Number (up to 12 digits)', value='987654321'),
     ],
-    color_map = {"correct": "green", "wrong": "red"}
     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),