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: [[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]]
|
|
|
|
|
|
|
|
|
|