.

app.py

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