cisco-ehsan commited on
Commit
a58ba75
·
verified ·
1 Parent(s): d53eed6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -4,7 +4,6 @@ This is a ModernBERT model fine-tuned for **Code Vulnerability Detection**.
4
  It is built on top of **SecureBERT 2.0**.
5
 
6
  ## Model Details
7
- - Original checkpoint: `/teamspace/studios/this_studio/secure_modern_bert/sentiment_classif/checkpoint_epoch_10.pth`
8
  - Model type: `classification`
9
  - Number of labels: 2
10
  - Architecture: ModernBertForSequenceClassification
@@ -13,8 +12,8 @@ It is built on top of **SecureBERT 2.0**.
13
  ## Usage Example
14
  ```python
15
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
16
- tokenizer = AutoTokenizer.from_pretrained('/teamspace/studios/this_studio/secure_modern_bert/Models/code-vuln-detection')
17
- model = AutoModelForSequenceClassification.from_pretrained('/teamspace/studios/this_studio/secure_modern_bert/Models/code-vuln-detection')
18
  ```
19
 
20
  ## Notes
 
4
  It is built on top of **SecureBERT 2.0**.
5
 
6
  ## Model Details
 
7
  - Model type: `classification`
8
  - Number of labels: 2
9
  - Architecture: ModernBertForSequenceClassification
 
12
  ## Usage Example
13
  ```python
14
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
15
+ tokenizer = AutoTokenizer.from_pretrained('CiscoAITeam/SecureBERT2.0-code-vuln-detection')
16
+ model = AutoModelForSequenceClassification.from_pretrained('CiscoAITeam/SecureBERT2.0-code-vuln-detection')
17
  ```
18
 
19
  ## Notes