Hugging Face's logo

Spaces:
Sher1988
/
caption-gen
Sleeping

App Files Community
caption-gen / utils /transforms.py
Sher1988's picture
Sher1988
Upload folder using huggingface_hub
38e36bb verified
Raw
History Blame Contribute Delete
101 Bytes
from torchvision.models import ResNet50_Weights
transforms = ResNet50_Weights.DEFAULT.transforms()