Update app.py

app.py

@@ -21,8 +21,6 @@ def remove_comments_from_code(code: str) -> str:
     code = re.sub(r'"""[\s\S]*?"""', '', code)
     code = re.sub(r"'''[\s\S]*?'''", '', code)
 
-    # 単一行コメント（# 以降を除去）
-    code = re.sub(r'#.*', '', code)
 
     return code
 
@@ -36,7 +34,7 @@ def get_query_and_candidates(seed: int = 8520):
     doc_str = query["docstring"]
 
     candidate_pool = [example for i, example in enumerate(dataset) if i != idx]
-    negatives = random.sample(candidate_pool, k=99)
+    negatives = random.sample(candidate_pool, k=19119)
     candidates = [correct_code] + [remove_comments_from_code(neg["code"]) for neg in negatives]  # 修正
     random.shuffle(candidates)