.

app.py

@@ -117,7 +117,7 @@ def predict_product(num1, num2):
         predicted_results_nocot = predicted_results_per_model['no']
         predicted_results_explicit_cot = predicted_results_per_model['explicit']
 
-        yield ground_truth_digits_digits, predicted_results_implicit, predicted_results_nocot, predicted_results_explicit_cot
+        yield ground_truth_results, predicted_results_implicit, predicted_results_nocot, predicted_results_explicit_cot
 
 color_map = {"correct": "green", "wrong": "red"}