Update CyberForge ML models and deployment artifacts

README.md

@@ -1,67 +1,62 @@
+
 ---
 license: mit
 tags:
-- cybersecurity
-- threat-detection
-- malware
-- phishing
-- anomaly-detection
-library_name: sklearn
-pipeline_tag: tabular-classification
+  - cybersecurity
+  - threat-detection
+  - phishing
+  - malware
+  - security
+language:
+  - en
 ---
 
-# CyberForge AI Security Models
+# CyberForge AI Models
 
-Production-ready machine learning models for real-time cybersecurity threat detection.
+Production-ready machine learning models for cybersecurity threat detection.
 
 ## Models Included
 
-| Model | Task | Accuracy | F1 Score | Inference |
-|-------|------|----------|----------|-----------|
-| phishing_detection | Detect phishing URLs | 98.9% | 0.989 | 0.02ms |
-| malware_detection | Identify malware | 99.8% | 0.998 | 0.001ms |
-| anomaly_detection | Network anomalies | 99.9% | 0.999 | 0.007ms |
-| web_attack_detection | Web attacks | 100% | 1.000 | 0.03ms |
+| Model | Task | Accuracy | F1 Score | Inference Time |
+|-------|------|----------|----------|----------------|
 
-## Quick Start
 
-```python
-from cyberforge_inference import CyberForgePredictor
+## Usage
+
+### Python
 
-# Initialize
-predictor = CyberForgePredictor()
+```python
+from inference import CyberForgeInference
 
-# Predict
-result = predictor.predict("phishing_detection", features)
-print(f"Threat: {result['prediction']}, Confidence: {result['confidence']}")
+inference = CyberForgeInference('./models')
+result = inference.predict('phishing_detection', features)
 ```
 
-## API Usage
+### API
 
 ```python
 import requests
 
 response = requests.post(
-    "https://your-api-endpoint/predict",
-    json={
-        "model": "phishing_detection",
-        "features": {...}
-    }
+    'https://huggingface.co/spaces/Che237/cyberforge/predict',
+    json={'model_name': 'phishing_detection', 'features': features}
 )
 ```
 
-## Features
-
-- **Real-time inference** < 1ms per prediction
-- **Multiple threat types**: Phishing, Malware, Anomalies, Web Attacks
-- **Production-ready**: Optimized for high-throughput
-- **Backend integration**: Compatible with Node.js/Python backends
+## Model Details
 
-## Training Data
+- **Framework**: scikit-learn
+- **Python Version**: 3.11+
+- **License**: MIT
 
-Models trained on 50,000+ samples from:
-- [CyberForge Datasets](https://huggingface.co/datasets/Che237/cyberforge-datasets)
+## Citation
 
-## License
+If you use these models, please cite:
 
-MIT License - Free for commercial and personal use.
+```bibtex
+@software{cyberforge2024,
+  title = {CyberForge AI Security Models},
+  year = {2024},
+  url = {https://huggingface.co/Che237/cyberforge-models}
+}
+```