requirements.txt

# Python dependencies for Few-Shot Bearing Fault Diagnosis with MLLMs
# Compatible with Python 3.8+

# Core dependencies
numpy>=1.24.0
pandas>=2.0.0
pyyaml>=6.0
openpyxl>=3.1.0

# Machine learning
scikit-learn>=1.3.0
scipy>=1.11.0

# OpenAI API (GPT-4o, GPT-5.1)
openai>=1.0.0

# Anthropic API (Claude 4.5 Haiku, Sonnet)
anthropic>=0.18.0

# Google Generative AI (optional, for Gemini models)
google-genai>=1.0.0

# Plotting and visualization
matplotlib>=3.7.0

# HTTP requests (for custom model endpoints)
requests>=2.31.0

# ============================================
# Optional: LLaVA Local Inference
# ============================================
# Uncomment the following if you want to run LLaVA-1.5-7B locally
# Requires CUDA-capable GPU

# torch>=2.0.0
# torchvision>=0.15.0
# transformers>=4.36.0
# accelerate>=0.25.0
# pillow>=10.0.0

# ============================================
# Optional: Prototypical Networks
# ============================================
# Uncomment the following if you want to run Prototypical Networks baseline
# Requires PyTorch and easyfsl

# torch>=2.0.0
# torchvision>=0.15.0
# easyfsl>=1.3.0
# pillow>=10.0.0