Shuu12121 commited on
Commit
066cc0b
·
verified ·
1 Parent(s): 182358f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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):