Fix check solution

app.py

@@ -1,5 +1,4 @@
 import re
-import os
 import spaces
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -20,7 +19,6 @@ eureka5_test_data = load_dataset(
     'gsarti/eureka-rebus', 'llm_sft',
     data_files=["id_test.jsonl", "ood_test.jsonl"],
     split = "train",
-    token=os.environ["HF_TOKEN"]
 )
 
 OUTPUTS_BASE_URL = "https://raw.githubusercontent.com/gsarti/verbalized-rebus/main/outputs/"
@@ -189,7 +187,7 @@ with gr.Blocks(css=custom_css) as demo:
                     
                     def check_solution(solution, correct_solution):
                         solution = unidecode(extract(solution))
-                        correct_solution = unidecode(extract(correct_solution, "h4"))
+                        correct_solution = unidecode(extract(correct_solution))
                         if solution == correct_solution:
                             gr.Info(gettext("CorrectSolutionMsg"))
                         else: