.

app.py

@@ -85,9 +85,10 @@ def predict_product(num1, num2):
                 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, ('0', 'wrong'))
+                if valid_input:
+                    if len([item for item in predicted_annotations_per_model[model_name] if item[1] is not None]) < len(ground_truth_digits_reversed):
+                        print ('jjere')
+                        predicted_annotations_per_model[model_name].insert(0, ('0', 'wrong'))
                 continue
                 
             generated_ids_per_model[model_name] = generated_ids