gradio==3.28.0
torch==2.1.0  # Or use 'torch==2.1.0+cu118' for CUDA-enabled GPU support
transformers
safetensors