Update app.py

app.py

@@ -193,7 +193,7 @@ with gr.Blocks(title="Beat the AI", theme=gr.themes.Base(), css=css) as demo:
         """
         # Will you beat the AI at these RAVEN puzzles?
 
-        *This demo features an early fine-tuned version of our forth-coming Idefics2 model. The model was fine-tuned on the [RAVEN](https://huggingface.co/datasets/HuggingFaceM4/RAVEN) dataset and reached 91% accuracy on the validation set.*
+        *This demo features an early fine-tuned version of our forth-coming Idefics2 model. The model was specifically fine-tuned on the [RAVEN](https://huggingface.co/datasets/HuggingFaceM4/RAVEN) dataset and reached 91% accuracy on the validation set.*
 
         To get started, load a new puzzle.
         """
@@ -243,8 +243,8 @@ with gr.Blocks(title="Beat the AI", theme=gr.themes.Base(), css=css) as demo:
             comparison_string = "You beat the AI!"
         elif user_pred != solution_letter and model_pred != solution_letter:
             comparison_string = "Both you and the AI got it wrong!"
-        else:
-            raise ValueError(model_pred, user_pred, solution)
+        elif user_pred != solution_letter and model_pred == solution_letter:
+            comparison_string = "The AI beat you!"
         
         return f"{win_or_loose} {comparison_string}\n{solution_string}"