mahdin70 commited on
Commit
7114df1
·
verified ·
1 Parent(s): 03ce0dc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -7
README.md CHANGED
@@ -85,7 +85,7 @@ with torch.no_grad():
85
  predicted_label = torch.argmax(predictions, dim=1).item()
86
 
87
  # Output the result
88
- print("⚠️ Vulnerable Code" if predicted_label == 1 else "Safe Code")
89
  ```
90
 
91
  ## Training Details
@@ -147,9 +147,4 @@ If you use this model in your research or applications, please cite:
147
 
148
  ## Model Card Authors
149
  - **Mukit Mahdin**
150
- - Contact: [[email protected]]
151
-
152
-
153
- ---
154
-
155
- Let me know if you need further modifications! 🚀
 
85
  predicted_label = torch.argmax(predictions, dim=1).item()
86
 
87
  # Output the result
88
+ print("Vulnerable Code" if predicted_label == 1 else "Safe Code")
89
  ```
90
 
91
  ## Training Details
 
147
 
148
  ## Model Card Authors
149
  - **Mukit Mahdin**
150
+ - Contact: [[email protected]]