elselse commited on
Commit
6519258
·
verified ·
1 Parent(s): b114361

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ classifier = pipeline(
10
  )
11
 
12
  # Load child-to-parent mapping
13
- with open("vulntrain/trainers/child_to_parent_mapping.json", "r") as f:
14
  child_to_parent = json.load(f)
15
 
16
  def predict_cwe(commit_message: str):
 
10
  )
11
 
12
  # Load child-to-parent mapping
13
+ with open("child_to_parent_mapping.json", "r") as f:
14
  child_to_parent = json.load(f)
15
 
16
  def predict_cwe(commit_message: str):