Spaces:

Shuu12121
/

CodeSearch-ModernBERT-Owl-Demo

Sleeping

Shuu12121 commited on Apr 16

Commit

066cc0b

verified ·

1 Parent(s): 182358f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ model = SentenceTransformer("Shuu12121/CodeSearch-ModernBERT-Owl")
 model.eval()
 # --- Load CodeSearchNet dataset (test split only) ---
-dataset = load_dataset("code_x_glue_tc_nl_code_search_adv", trust_remote_code=True)
 # --- Query & Candidate Generator ---
 def get_random_query(seed: int = 42):

 model.eval()
 # --- Load CodeSearchNet dataset (test split only) ---
+dataset = load_dataset("code_x_glue_tc_nl_code_search_adv", trust_remote_code=True, split="test")
 # --- Query & Candidate Generator ---
 def get_random_query(seed: int = 42):