requirements.txt

# Core dependencies
gradio==4.19.2
torch==2.2.0
torchvision==0.17.0
torchaudio==2.2.0
transformers==4.37.2
diffusers==0.25.0
accelerate==0.27.2

# NLP tools
spacy==3.7.2
sacremoses==0.0.53
sentencepiece==0.1.99
protobuf==4.25.2
tokenizers==0.15.2
ftfy==6.1.3
regex==2023.12.25

# Image processing
Pillow==10.2.0
opencv-python==4.9.0.80

# Deep learning utilities
numpy==1.26.4
scipy==1.12.0
tqdm==4.66.2
sympy==1.12
networkx==3.2.1

# HuggingFace specific
huggingface-hub==0.20.3
safetensors==0.4.2

# Additional utilities
requests==2.31.0
packaging==23.2
filelock==3.13.1
psutil==5.9.8
typing_extensions==4.9.0
pyyaml==6.0.1

# Optional - for better performance with NVIDIA GPUs
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.2.0+cu121
torchvision==0.17.0+cu121
torchaudio==2.2.0+cu121