.

app.py

@@ -83,7 +83,10 @@ def predict_product(num1, num2):
             
             if next_token_id.item() == tokenizer.eos_token_id:
                 finished_per_model[model_name] = True
+                print (step)
+                print (len(ground_truth_digits_reversed))
                 if step < len(ground_truth_digits_reversed): # stopped earlier
+                    print ('jere')
                     predicted_annotations_per_model[model_name].insert(0, (' ', 'wrong'))
                 continue