saving-willy-dev / requirements.txt
rmm
fix: get requirements where HF build expects them
a7ece71
raw
history blame
502 Bytes
numpy==1.24
pandas==2.2.3
# frontend
streamlit==1.39.0
## for nice map tiles
folium==0.18.0
streamlit_folium==0.23.1
# backend
datasets==3.0.2
# running ML models
## to use ML models hosted on HF
huggingface-hub==0.26.1
transformers==4.46.0
#torch
## +minimal reqs for locally running model - with PIL load
pytorch-lightning<=1.5.10
timm==0.5.4
pillow==10.4.0
## also need opencv for handling images in the later model with preprocessing
opencv-python-headless==4.5.5.64
albumentations==1.1.0