- Use runpod/pytorch:1.0.3-cu1290-torch260-ubuntu2204 base image
- Torch 2.6.0 required by transformers for secure model loading
- CUDA 12.9 compatible
- Downloaded 191K DDI pairs from TDC DrugBank
- Fetched 1,634 drug names from PubChem API (96% hit rate)
- Created complete training dataset with:
- Real drug names (not just IDs)
- 86 interaction type descriptions
- Severity labels (minor/moderate/major/contraindicated)
- Bundled 34MB data file in Docker image
- Handler loads real data instead of curated samples