Update app.py

app.py

@@ -91,7 +91,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # Filter the dataset to include ONLY the target task ID
     # This uses the 'filter' method available on Hugging Face datasets.
-    subset = dataset.filter(lambda example: example['task_id'] == target_id)
+    subset = dataset.filter(lambda example: example['task_id'] in target_task_ids)
     
     results_log = []
     answers_payload = []