.

app.py

@@ -39,8 +39,12 @@ def predict_product(num1, num2):
     except ValueError:
         valid_input = False
     if valid_input:
-        correct_product = str(num1_int * num2_int)
-        is_correct = (prediction == correct_product)
+        correct_product = num1_int * num2_int
+        try:
+            prediction = int(prediction)
+            is_correct = (prediction == correct_product)
+        except ValueError:
+            is_correct = False
         result_color = "green" if is_correct else "red"
         result_message = "Correct!" if is_correct else f"Incorrect! The correct product is {correct_product}."
     else: