Update app.py

app.py

@@ -20,7 +20,7 @@ def get_random_query(lang: str, seed: int = 42):
     random.seed(seed)
     idx = random.randint(0, len(subset) - 1)
     sample = subset[idx]
-    return sample["function"] or "", sample["docstring"] or ""
+    return sample["func_code_string"] or "", sample["func_documentation_string"] or ""
 
 @GPU
 def code_search_demo(lang: str, seed: int):
@@ -29,7 +29,7 @@ def code_search_demo(lang: str, seed: int):
 
     # ランダムに取得した同一言語の10件の関数とドキュメントを比較対象として選択
     candidates = dataset_all.filter(lambda x: x["language"] == lang).shuffle(seed=seed).select(range(10))
-    candidate_texts = [c["function"] or "" for c in candidates]
+    candidate_texts = [c["func_code_string"] or "" for c in candidates]
     candidate_embeddings = model.encode(candidate_texts, convert_to_tensor=True)
 
     # 類似度計算