update

app.py

@@ -150,8 +150,7 @@ def calculate_order_by_first_substring(selected_models):
         lambda x: x["parsed_judge_response"].eq(1).all()
     )
 
-    query_ids = query_ids_df.query_id.unique()
-    fsm_ids = query_ids_df.fsm_id.unique()
+
 
     text_only = all_data[all_data["Model Type"] == "Text Only"]
     text_only_filtered = text_only[text_only["fsm_id"].isin(fsm_ids)]
@@ -169,6 +168,9 @@ def calculate_order_by_first_substring(selected_models):
     )
     text_only_filtered.sort_values("Accuracy", ascending=False, inplace=True)
 
+    query_ids = text_only_filtered.query_id.unique()
+    fsm_ids = text_only_filtered.fsm_id.unique()
+
     number_of_queries = len(query_ids)
     number_of_fsms = len(fsm_ids)