fixed (#5)

- fixed (398beaa025adef597538b85b2f682e7d611860b2)


Co-authored-by: Oleksandr Lukashov <[email protected]>

interfaces/compare_pipeline.py

@@ -100,14 +100,18 @@ def compute_scores(*args):
 def compute_table(*args):
     gliclass_results, st_results = compute_scores(*args)
     max_docs = int(os.getenv("MAX_DOCS"))
+    labels = args[1:]
+
     gliclass_labels = gliclass_results[:max_docs]
     st_labels = st_results[:max_docs]
+
+    label_rank_gliclass = {label: rank + 1 for rank, label in enumerate(gliclass_labels) if label}
+    label_rank_st = {label: rank + 1 for rank, label in enumerate(st_labels) if label}
     df = pd.DataFrame({
-        "Rank": list(range(1, max_docs + 1)),
-        "GLiClass Label": gliclass_labels,
-        "CrossEncoder Label": st_labels,
+        "Document": labels,
+        "GLiClass Rank": [label_rank_gliclass.get(label, "") for label in labels],
+        "Cross-Encoder Rank": [label_rank_st.get(label, "") for label in labels],
     })
-
     return df
 
 examples = [
@@ -116,13 +120,14 @@ examples = [
 ]
 
 with gr.Blocks(title="GLiClass-Reranker") as compare_pipeline:
+    example_state = gr.State(value=examples)
     inputs = []
     query = gr.Textbox(
         value=examples[0][0], label="Text query", placeholder="Enter your query here", lines=4
     )
-    labels = [gr.Textbox(value=label, label=f"Label {i+1}") for i, label in enumerate(examples[0][1:])]
+    labels = [gr.Textbox(value=label, label=f"Document {i+1}") for i, label in enumerate(examples[0][1:])]
     submit_btn = gr.Button("Compare")
-    result_table = gr.Dataframe(headers=["Rank", "GLiClass Label", "CrossEncoder Label"],
+    result_table = gr.Dataframe(headers=["Document", "GLiClass Rank", "Cross-Encoder Rank"],
                                 label="Comparison Table",
                                 interactive=False)
 

interfaces/scores_pipeline.py

@@ -86,6 +86,7 @@ def classification(*args) -> List[str]:
     return docs + scores
 
 with gr.Blocks(title="GLiClass-Reranker") as scores_pipeline:
+    example_state = gr.State(value=examples)
     inputs = []
     outputs = []
     query = gr.Textbox(