Image2Body_gradio / requirements.txt
yeq6x's picture
test
4ae6aa0
raw
history blame
532 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.2.0
torchvision==0.17.0
torchaudio==2.2.0
transformers
diffusers==0.27.0 # pth file cannot be loaded in the latest version
Flask==3.0.3
Flask-Cors==4.0.0
Flask-SocketIO==5.3.6
gradio==5.5.0
huggingface-hub==0.25.2
kornia==0.7.1
numpy==1.23.5
opencv-python==4.9.0.80
Pillow==10.3.0
Requests==2.32.3
tensorflow==2.16.1
transforms==0.2.1
tokenizers
pytorch_lightning
python-dotenv
peft==0.11.1
flask_limiter==3.7.0
gevent==24.2.1
gevent-websocket==0.10.1
firebase-admin