text-guided-image-colorization / requirements.txt
LogicGoInfotechSpaces's picture
Integrate CCO colorization models (eccv16 and siggraph17) - Add CCO colorizers module from kinsung/cco - Update /colorize endpoint to support model selection parameter - Add scikit-image dependency - Maintain backward compatibility with existing GAN model - Update MongoDB logging to track model type used
b6181ba
raw
history blame contribute delete
217 Bytes
torch
torchvision
fastapi
uvicorn
gradio
pillow
firebase-admin
fastai
huggingface_hub
pydantic-settings
opencv-python
numpy
accelerate
transformers
diffusers
safetensors
ftfy
httpx
email-validator
pymongo
scikit-image