Update README.md
Browse files
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("
|
| 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: [mahdin.mukit248@gmail.com]
|
| 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: [mahdin.mukit248@gmail.com]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|