mahdin70
/

unixcoder-code-vulnerability-detector

Text Classification

Model card Files Files and versions

mahdin70 commited on Mar 1

Commit

8fc8fb1

·

verified ·

1 Parent(s): 436bcbb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -60,7 +60,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 # Load the fine-tuned model
-tokenizer = AutoTokenizer.from_pretrained("mahdin70/unixcoder-code-vulnerability-detector")
 model = AutoModelForSequenceClassification.from_pretrained("mahdin70/unixcoder-code-vulnerability-detector")
 # Sample code snippet

 import torch
 # Load the fine-tuned model
+tokenizer = AutoTokenizer.from_pretrained("microsoft/unixcoder-base")
 model = AutoModelForSequenceClassification.from_pretrained("mahdin70/unixcoder-code-vulnerability-detector")
 # Sample code snippet