Pranav Pc commited on
Commit
2e08b23
·
1 Parent(s): 98f1ee4

Align model path with actual file

Browse files
Files changed (1) hide show
  1. src/inference.py +1 -1
src/inference.py CHANGED
@@ -10,7 +10,7 @@ sys.path.append(str(Path(__file__).parent.parent.parent))
10
  from src.model import VulnerabilityCodeT5
11
 
12
  class VulnerabilityDetector:
13
- def __init__(self, model_path="models/best_model.pt",
14
  model_name="Salesforce/codet5-base", max_length=256):
15
 
16
  ### CHANGED FOR DEPLOYMENT
 
10
  from src.model import VulnerabilityCodeT5
11
 
12
  class VulnerabilityDetector:
13
+ def __init__(self, model_path="models/best_model_clean.pt",
14
  model_name="Salesforce/codet5-base", max_length=256):
15
 
16
  ### CHANGED FOR DEPLOYMENT