Update README.md
Browse files
README.md
CHANGED
|
@@ -46,7 +46,7 @@ import torch
|
|
| 46 |
|
| 47 |
# Download model
|
| 48 |
model_path = hf_hub_download(
|
| 49 |
-
repo_id="
|
| 50 |
filename="pytorch_model.pth"
|
| 51 |
)
|
| 52 |
|
|
@@ -69,11 +69,11 @@ checkpoint = torch.load(model_path, map_location='cpu')
|
|
| 69 |
|
| 70 |
## Citation
|
| 71 |
```bibtex
|
| 72 |
-
@misc{
|
| 73 |
-
author =
|
| 74 |
-
title =
|
| 75 |
-
year =
|
| 76 |
-
publisher =
|
| 77 |
-
url =
|
| 78 |
}
|
| 79 |
```
|
|
|
|
| 46 |
|
| 47 |
# Download model
|
| 48 |
model_path = hf_hub_download(
|
| 49 |
+
repo_id="seffyehl/BetterShield",
|
| 50 |
filename="pytorch_model.pth"
|
| 51 |
)
|
| 52 |
|
|
|
|
| 69 |
|
| 70 |
## Citation
|
| 71 |
```bibtex
|
| 72 |
+
@misc{bettershield-2025,
|
| 73 |
+
author = Orion,
|
| 74 |
+
title = Hate Speech Detection with Rationale Extraction,
|
| 75 |
+
year = 2025,
|
| 76 |
+
publisher = HuggingFace,
|
| 77 |
+
url = https://huggingface.co/seffyehl/BetterShield
|
| 78 |
}
|
| 79 |
```
|