.

app.py

@@ -91,15 +91,15 @@ 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:
+                if '=' not in predicted_digits_reversed:
+                    print ('aaa')
                     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:]
+                    print ('bbbaaa')
+                    equal_sign_position = predicted_digits_reversed.index('=')
+                    predicted_annotations = [(predicted_digit, None) for predicted_digit in predicted_digits_reversed[:equal_sign_position+1]]
+                    predicted_digits_reversed = predicted_digits_reversed[equal_sign_position+1:]
 
             for i in range(len(predicted_digits_reversed)):
                 predicted_digit = predicted_digits_reversed[i]