Pranav Pc commited on
Commit ·
2e08b23
1
Parent(s): 98f1ee4
Align model path with actual file
Browse files- 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/
|
| 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
|