phronesis-ml-endpoint / requirements.txt
baliddeki's picture
Upload 5 files
79c27a2 verified
raw
history blame
323 Bytes
# Core dependencies
torch==2.0.1
torchvision==0.15.2
transformers==4.44.2
gradio==5.0
numpy==1.26.2
Pillow==10.0.1
fastapi
# Additional dependencies
huggingface_hub==0.25.1 # Compatible with both transformers and gradio
torchmetrics==1.5.1
nltk==3.8.1
scikit-learn==1.3.0
tqdm==4.66.1
sentencepiece==0.1.99
pydicom==2.4.1