Add pretrained SAGENET checkpoints

README.md

@@ -1,3 +1,58 @@
 ---
 license: apache-2.0
+tags:
+  - protocol-reverse-engineering
+  - network-security
+  - deep-learning
+  - pytorch
+language:
+  - en
 ---
+
+# SAGENET Pre-trained Models
+
+Pre-trained models for **SAGENET** - a protocol reverse engineering tool for format inference using deep learning.
+
+## Model Overview
+
+| Model | Protocol | Size |
+|-------|----------|------|
+| `arp_best.pth` | ARP | 90MB |
+| `dns_best.pth` | DNS | 90MB |
+| `icmp_best.pth` | ICMP | 90MB |
+| `mbtcp_best.pth` | Modbus TCP | 90MB |
+| `nbns_best.pth` | NBNS | 90MB |
+| `ntp_best.pth` | NTP | 90MB |
+| `s7comm_best.pth` | S7comm | 90MB |
+| `smb_best.pth` | SMB | 90MB |
+| `tcp_best.pth` | TCP | 90MB |
+| `udp_best.pth` | UDP | 90MB |
+
+## Quick Start
+
+```python
+from huggingface_hub import hf_hub_download
+
+# Download a specific model
+model_path = hf_hub_download(
+    repo_id="Prescot12/SAGENET",
+    filename="checkpoints/tcp_best.pth"
+)
+```
+
+## Usage
+
+See the [main repository](https://github.com/infinityay/SAGENET) for complete usage instructions.
+
+```bash
+# Test with pre-trained models
+python main.py --mode test --protocols tcp udp dns
+```
+
+## Citation
+
+If you use these models in your research, please cite our work.
+
+## License
+
+Apache 2.0

checkpoints/arp_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4cad7b20e8dd4b3f8b556e0bad2c04c0b4879bb778637cbdd02ab575ab750775
+size 94027428

checkpoints/dns_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3be58892f7d7f65919802e3e9c467070362019336120a17198629fd2a63d8419
+size 94027428

checkpoints/icmp_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ccff7e5cce02983fc17106d7fb8ff68d732e980c5f9d0dc44e80c1de818526f8
+size 94028090

checkpoints/mbtcp_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:96270e6a248858c76354441772f82e93d5110330d587ab8636e383647d2cf0f2
+size 94028752

checkpoints/nbns_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d96b26ebf26a2fc78233ddb7ccd7e92b17253b3d17d6992f391c8ebaeaa2bb7d
+size 94028090

checkpoints/ntp_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:654a4ece94c90fc7d185a96f1af74e8bbaf75ce034cd2087cfe453469a8d9c79
+size 94027428

checkpoints/s7comm_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:537ca87c531b53f76703bddf856e5464225eb7a47940d35f7be645c4b6c37d15
+size 94029414

checkpoints/smb_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3aee1f3e50facc24d6bcf29f97665da4a3626eab42fa8f8e76f2b4c80b372099
+size 94027428

checkpoints/tcp_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a5af96b35f09936c80c476380663de083ec792138cea95c7756c20305331673
+size 94027428

checkpoints/udp_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da669f0b20c588c62654347845f361c03694043740ebab7c09b51c036c446c6c
+size 94027428