Upload folder using huggingface_hub

README.md

@@ -45,14 +45,22 @@ C2Sentinel analyzes network connection patterns to identify C2 beacon activity.
 ## Installation
 
 ```bash
-pip install torch numpy safetensors
+pip install torch numpy safetensors huggingface_hub
 ```
 
 ---
 
 ## Usage
 
-### Loading the Model
+### Loading from HuggingFace Hub
+
+```python
+from c2sentinel import C2Sentinel
+
+sentinel = C2Sentinel.from_pretrained('danielostrow/c2sentinel')
+```
+
+### Loading from Local Files
 
 ```python
 from c2sentinel import C2Sentinel

c2sentinel.py

@@ -1753,6 +1753,47 @@ class C2Sentinel:
 
         return cls(model, config, device)
 
+    @classmethod
+    def from_pretrained(cls, repo_id: str, device: str = 'auto', cache_dir: Optional[str] = None) -> 'C2Sentinel':
+        """
+        Load model from HuggingFace Hub.
+
+        Args:
+            repo_id: HuggingFace repository ID (e.g., 'danielostrow/c2sentinel')
+            device: Device to load model on ('auto', 'cpu', 'cuda', 'mps')
+            cache_dir: Optional cache directory for downloaded files
+
+        Returns:
+            Loaded C2Sentinel instance
+        """
+        try:
+            from huggingface_hub import hf_hub_download
+        except ImportError:
+            raise ImportError("huggingface_hub is required for from_pretrained. Install with: pip install huggingface_hub")
+
+        # Download model files
+        model_path = hf_hub_download(
+            repo_id=repo_id,
+            filename="c2_sentinel.safetensors",
+            cache_dir=cache_dir
+        )
+        config_path = hf_hub_download(
+            repo_id=repo_id,
+            filename="c2_sentinel.json",
+            cache_dir=cache_dir
+        )
+
+        # Load config
+        with open(config_path, 'r') as f:
+            config = C2SentinelConfig.from_dict(json.load(f))
+
+        # Load model
+        model = LogBERTC2Sentinel(config)
+        state_dict = load_file(str(model_path))
+        model.load_state_dict(state_dict)
+
+        return cls(model, config, device)
+
     @classmethod
     def create_new(cls, device: str = 'auto') -> 'C2Sentinel':
         """Create a new untrained model."""