suidinpa
/

bottle-anomaly-detection

Model card Files Files and versions

bottle-anomaly-detection / main.py

suidinpa's picture

Create main.py

9e5096b verified over 1 year ago

history blame contribute delete

149 Bytes

	from anomalib.models import AnomalyModel

	model = AnomalyModel()
	model.load_state_dict(torch.load('model.bin'))

	model.save_pretrained('./my_model')