Update app.py

app.py

@@ -21,8 +21,8 @@ from content import format_error, format_warning, format_log, TITLE, INTRODUCTIO
 TOKEN = os.environ.get("TOKEN", None)
 
 OWNER="Blanca"
-DATA_DATASET = f"{OWNER}/CQs-Gen_test"
-INTERNAL_DATA_DATASET = f"{OWNER}/CQs-Gen_test" # TODO: change to the dataset that contains the embeddings
+DATA_DATASET = f"{OWNER}/CQs-Gen_test_embeddings"
+INTERNAL_DATA_DATASET = f"{OWNER}/CQs-Gen_test_embeddings
 SUBMISSION_DATASET = f"{OWNER}/submissions_internal"
 SUBMISSION_DATASET_PUBLIC = f"{OWNER}/submissions_public"
 #CONTACT_DATASET = f"{OWNER}/contact_info" # TODO: I should reactivate this
@@ -56,7 +56,7 @@ test_results = load_dataset(
     verification_mode=VerificationMode.NO_CHECKS,
     trust_remote_code=True,
 )
-eval_results = {"test": test_results}
+eval_results['test'] = load_dataset(RESULTS_DATASET, YEAR_VERSION, token=TOKEN, download_mode="force_redownload", verification_mode=VerificationMode.NO_CHECKS, trust_remote_code=True)
 # TODO: I should reactivate saving contact infos
 #contact_infos = load_dataset(CONTACT_DATASET, YEAR_VERSION, token=TOKEN, download_mode="force_redownload", verification_mode=VerificationMode.NO_CHECKS, trust_remote_code=True)
 def get_dataframe_from_results(eval_results, split):
@@ -220,6 +220,8 @@ def add_new_eval(
             for id_to_eval, line in data.items():
                 intervention_score = 0
                 for indx, intervention_id in enumerate(gold_dataset['intervention_id']):
+                    if indx == 3:
+                        break
                     if id_to_eval == intervention_id:
                         references = gold_dataset['cqs']
                         reference_set = [row['cq'] for row in references[indx]]