.

app.py

@@ -14,7 +14,7 @@ no_cot_model = AutoModelForCausalLM.from_pretrained(no_cot_model_name)
 explicit_cot_model_name = 'yuntian-deng/gpt2-explicit-cot-multiplication'
 explicit_cot_model = AutoModelForCausalLM.from_pretrained(explicit_cot_model_name)
 
-models = {'implicit': implicit_cot_model_name, 'no': no_cot_model, 'explicit': explicit_cot_model}
+models = {'implicit': implicit_cot_model, 'no': no_cot_model, 'explicit': explicit_cot_model}
 
 # Constants
 MAX_PRODUCT_DIGITS_PER_MODEL = {'implicit': 100, 'no': 100, 'explicit': 900}