.

app.py

@@ -90,6 +90,17 @@ def predict_product(num1, num2):
         
             predicted_annotations = []
             is_correct_sofar = True
+            if model_name == 'implicit':
+                num_equal_signs = sum([1 for token in predicted_digits_reversed if token == '='])
+                if num_equal_signs < 2:
+                    predicted_annotations = [(predicted_digit, None) for predicted_digit in predicted_digits_reversed]
+                    predicted_digits_reversed = []
+                else:
+                    first_equal_sign_position = predicted_digits_reversed.index('=')
+                    second_equal_sign_position = predicted_digits_reversed[first_equal_sign_position+1:].index('=') + first_equal_sign_position+1
+                    predicted_annotations = [(predicted_digit, None) for predicted_digit in predicted_digits_reversed[:second_equal_sign_position+1]]
+                    predicted_digits_reversed = predicted_digits_reversed[second_equal_sign_position+1:]
+
             for i in range(len(predicted_digits_reversed)):
                 predicted_digit = predicted_digits_reversed[i]
                 if i >= len(ground_truth_digits_reversed):