Update README.md
Browse files
README.md
CHANGED
|
@@ -38,7 +38,7 @@ THREAT_CLASSES = {
|
|
| 38 |
image = Image.open("Path_to_image")
|
| 39 |
|
| 40 |
# pre-trained weights
|
| 41 |
-
weights_url = "https://huggingface.co/Subh775/Threat-Detection-
|
| 42 |
weights_filename = "checkpoint_best_total.pth"
|
| 43 |
|
| 44 |
# Download weights if not already present
|
|
|
|
| 38 |
image = Image.open("Path_to_image")
|
| 39 |
|
| 40 |
# pre-trained weights
|
| 41 |
+
weights_url = "https://huggingface.co/Subh775/Threat-Detection-RF-DETR/resolve/main/checkpoint_best_total.pth"
|
| 42 |
weights_filename = "checkpoint_best_total.pth"
|
| 43 |
|
| 44 |
# Download weights if not already present
|