docs: remove test_app references, update ONNX count to 5 variants
Browse files
AGENTS.md
CHANGED
|
@@ -9,8 +9,7 @@ Hugging Face model repo for `buildborderless/CommunityForensics-DeepfakeDet-ViT`
|
|
| 9 |
- **`preprocessor_config.json`** — CLIP-style normalization, resize to 440, center-crop to 384
|
| 10 |
- **`modeling_vit_classifier.py`** — **DEPRECATED** (moved to `scripts/`). Use standard HF path below.
|
| 11 |
- **`pretrained_weights/`** — original `.pt` checkpoints from training (also LFS)
|
| 12 |
-
- **`onnx/`** —
|
| 13 |
-
- **`test_app/`** — local Gradio app for testing PyTorch vs ONNX, benchmarking, and comparison
|
| 14 |
|
| 15 |
## Usage
|
| 16 |
The model is hosted on Hugging Face. The standard way to load it is via `transformers`:
|
|
|
|
| 9 |
- **`preprocessor_config.json`** — CLIP-style normalization, resize to 440, center-crop to 384
|
| 10 |
- **`modeling_vit_classifier.py`** — **DEPRECATED** (moved to `scripts/`). Use standard HF path below.
|
| 11 |
- **`pretrained_weights/`** — original `.pt` checkpoints from training (also LFS)
|
| 12 |
+
- **`onnx/`** — 5 pre-exported ONNX variants (16MB–84MB) for CPU/GPU deployment. See README for variant guide.
|
|
|
|
| 13 |
|
| 14 |
## Usage
|
| 15 |
The model is hosted on Hugging Face. The standard way to load it is via `transformers`:
|
README.md
CHANGED
|
@@ -143,19 +143,6 @@ logits = output[0][0] # shape [2]: [real_score, fake_score]
|
|
| 143 |
|
| 144 |
## Local Test App
|
| 145 |
|
| 146 |
-
A local Gradio app for testing PyTorch and ONNX models side by side with benchmarks:
|
| 147 |
-
|
| 148 |
-
```bash
|
| 149 |
-
git clone https://huggingface.co/buildborderless/CommunityForensics-DeepfakeDet-ViT
|
| 150 |
-
cd CommunityForensics-DeepfakeDet-ViT
|
| 151 |
-
pip install -r test_app/requirements.txt
|
| 152 |
-
python test_app/app.py
|
| 153 |
-
```
|
| 154 |
-
|
| 155 |
-
Three tabs: PyTorch vs ONNX comparison, Benchmark (all 8 ONNX variants on test images), and help.
|
| 156 |
-
|
| 157 |
-
---
|
| 158 |
-
|
| 159 |
## Citation
|
| 160 |
|
| 161 |
```bibtex
|
|
|
|
| 143 |
|
| 144 |
## Local Test App
|
| 145 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 146 |
## Citation
|
| 147 |
|
| 148 |
```bibtex
|