Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ model = SentenceTransformer("Shuu12121/CodeSearch-ModernBERT-Owl")
|
|
11 |
model.eval()
|
12 |
|
13 |
# --- Load CodeSearchNet dataset (test split only) ---
|
14 |
-
dataset = load_dataset("code_x_glue_tc_nl_code_search_adv", trust_remote_code=True)
|
15 |
|
16 |
# --- Query & Candidate Generator ---
|
17 |
def get_random_query(seed: int = 42):
|
|
|
11 |
model.eval()
|
12 |
|
13 |
# --- Load CodeSearchNet dataset (test split only) ---
|
14 |
+
dataset = load_dataset("code_x_glue_tc_nl_code_search_adv", trust_remote_code=True, split="test")
|
15 |
|
16 |
# --- Query & Candidate Generator ---
|
17 |
def get_random_query(seed: int = 42):
|