Image_Captioning_Ensemble_API / requirements.txt
SaniaE's picture
added requirement.txt
9abcde9 verified
raw
history blame contribute delete
294 Bytes
fastapi
uvicorn
python-multipart
torch==2.2.1
torchvision==0.17.1
transformers==4.40.0
accelerate==0.29.1
datasets==2.18.0
evaluate==0.4.1
huggingface-hub==0.21.4
sentence-transformers==3.0.1
pandas==2.2.1
numpy==1.26.4
pillow==10.2.0
matplotlib==3.9.1
scikit-learn==1.4.1.post1
scipy==1.12.0