Spaces:

Dharshaneshwaran
/

deepfake

Running

deepfake / model.py

Dharshaneshwaran

Full updated code with finding ai generated images too

ddcedb5 2 months ago

152 Bytes

	from torchvision.models import efficientnet_v2_s, EfficientNet_V2_S_Weights

	model = efficientnet_v2_s(weights=EfficientNet_V2_S_Weights.IMAGENET1K_V1)