18 digits

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load models
-implicit_cot_model_name = 'yuntian-deng/gpt2-implicit-cot-multiplication'
+implicit_cot_model_name = 'yuntian-deng/gpt2-implicit-cot-multiplication-18-digits'
 implicit_cot_model = AutoModelForCausalLM.from_pretrained(implicit_cot_model_name)
 tokenizer = AutoTokenizer.from_pretrained(implicit_cot_model_name)
 
@@ -144,8 +144,8 @@ color_map = {"correct": "green", "wrong": "red"}
 demo = gr.Interface(
     fn=predict_product,
     inputs=[
-        gr.Textbox(label='First Number (up to 15 digits)', value='123456789'),
-        gr.Textbox(label='Second Number (up to 15 digits)', value='987654321'),
+        gr.Textbox(label='First Number (up to 18 digits)', value='123456789'),
+        gr.Textbox(label='Second Number (up to 18 digits)', value='987654321'),
     ],
     outputs=[
         gr.HighlightedText(label='Ground Truth Product', combine_adjacent=False, show_legend=False, color_map=color_map),