Agent_Space / requirements.txt
arout2026's picture
Upload 16 files
af3b3c9 verified
# ============================================================================
# OPTIMAX SECURITY AGENT - REQUIREMENTS
# Version: 3.1.0 (Multi-Model AI + Anomaly Detection)
# ============================================================================
# Core Framework - Gradio 6.3.0 Compatible
gradio==6.3.0
fastapi==0.115.5
uvicorn[standard]==0.32.1
pydantic==2.10.5
# AI/ML Dependencies - Transformers & PyTorch
transformers==4.46.0
torch==2.5.1
numpy<2.0.0
sentencepiece==0.1.99
accelerate==0.34.2
# Machine Learning Libraries - Includes Isolation Forest for Anomaly Detection
scikit-learn==1.5.2 # ✅ Provides IsolationForest (no additional install needed!)
scipy==1.14.1
# Salesforce Integration
simple-salesforce==1.12.4
requests==2.31.0
# Utilities
python-dotenv==1.0.0
# ============================================================================
# NOTES:
# ============================================================================
#
# ✅ No changes needed from previous version!
#
# scikit-learn==1.5.2 already includes:
# - IsolationForest (anomaly detection)
# - cosine_similarity (used by AI models)
# - All necessary ML utilities
#
# Python Version: 3.10, 3.11, or 3.13
# Tested on: Hugging Face Spaces (Python 3.11)
#
# Total Models:
# 1. CodeBERT (125M) - from transformers
# 2. RoBERTa (125M) - from transformers
# 3. SecBERT (110M) - from transformers
# 4. Isolation Forest - from scikit-learn ✅ NEW!
#
# ============================================================================