.

app.py

@@ -55,7 +55,7 @@ def predict_product(num1, num2):
     for step in range(max(MAX_PRODUCT_DIGITS_PER_MODEL.values())):  # Set a maximum limit to prevent infinite loops
         # Ground Truth
         if not valid_input:
-            ground_truth_annotations = []
+            ground_truth_annotations = [('Invalid Input!', None)]
         else:
             ground_truth_annotations = [(ground_truth_digit, None) for ground_truth_digit in ground_truth_digits_reversed[:step+1]]
             ground_truth_annotations = ground_truth_annotations[::-1]