fix: gradio 不支持raw_labels

app.py

@@ -22,11 +22,12 @@ def generate(query1, query2, source_code):
         source_code = "# Use the built-in enumerator\nfor idx, x in enumerate(xs):\n    print(idx, x)"
 
     results = []
-    for model in [model1, model2, model3, model4, model5]:
+    model_names = ["jinaai/jina-embeddings-v2-base-code", "jinaai/jina-embeddings-v2-base-en", "jinaai/jina-embeddings-v2-base-zh", "aspire/acge_text_embedding", "intfloat/multilingual-e5-large"]
+    for model, name in zip([model1, model2, model3, model4, model5], model_names):
         embeddings = model.encode([query1, query2, source_code])
         score1 = cos_sim(embeddings[0], embeddings[2])
         score2 = cos_sim(embeddings[1], embeddings[2])
-        results.append((float(score1), float(score2)))
+        results.append([name, float(score1), float(score2)])
 
     return results
 
@@ -39,9 +40,8 @@ gr.Interface(
     ],
     outputs=[
         gr.Dataframe(
-            headers=["Query1 Score", "Query2 Score"],
+            headers=["Model", "Query1 Score", "Query2 Score"],
             label="Similarity Scores",
-            row_labels=["jinaai/jina-embeddings-v2-base-code", "jinaai/jina-embeddings-v2-base-en", "jinaai/jina-embeddings-v2-base-zh", "aspire/acge_text_embedding", "intfloat/multilingual-e5-large"]
         )
     ],
 ).launch()