Update app.py

app.py

@@ -194,16 +194,18 @@ 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.*
+
+        To get started, load a new puzzle.
         """
     )
-    load_new_sample = gr.Button(value="Load new sample")
+    load_new_sample = gr.Button(value="Load a new puzzle")
     with gr.Row(equal_height=True):
         with gr.Column(scale=4, min_width=250) as upload_area:
             imagebox = gr.Image(
                 image_mode="L",
                 type="pil",
                 visible=True,
-                sources=None,
+                sources=[],
             )
         with gr.Column(scale=4):
             with gr.Row():
@@ -231,7 +233,7 @@ with gr.Blocks(title="Beat the AI", theme=gr.themes.Base(), css=css) as demo:
 
     def result_string(model_pred, user_pred, solution):
         solution_letter = chr(ord('A') + int(solution))
-        solution_string = f"The correct answer is f{solution_letter}."
+        solution_string = f"The correct answer is {solution_letter}."
         
         win_or_loose = "🥇" if user_pred == solution_letter else "🙈"
         
@@ -241,6 +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)
         
         return f"{win_or_loose} {comparison_string}\n{solution_string}"