Update app.py

app.py

@@ -190,7 +190,11 @@ h1{display: flex;align-items: center;justify-content: center;gap: .25em}
 
 with gr.Blocks(title="Beat the AI", theme=gr.themes.Base(), css=css) as demo:
     gr.Markdown(
-        "Are you smarter than the AI?"
+        """
+        # 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.*
+        """
     )
     load_new_sample = gr.Button(value="Load new sample")
     with gr.Row(equal_height=True):
@@ -226,7 +230,7 @@ with gr.Blocks(title="Beat the AI", theme=gr.themes.Base(), css=css) as demo:
                 result.render()
 
     def result_string(model_pred, user_pred, solution):
-        solution_letter = chr(ord('A') + int(z))
+        solution_letter = chr(ord('A') + int(solution))
         solution_string = f"The correct answer is f{solution_letter}."
         
         win_or_loose = "🥇" if user_pred == solution_letter else "🙈"