Sathya77
/

ViT_MvTec

Image Classification

Model card Files Files and versions

Sathya77 commited on Sep 30

Commit

a4a4abc

·

verified ·

1 Parent(s): 2284d25

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -72,7 +72,7 @@ from transformers import ViTModel
 import joblib, numpy as np, torch
 # Load model + embeddings
-model = ViTModel.from_pretrained("your-username/ViT_MvTec")
 knn = joblib.load("knn.pkl")
 train_embeddings = np.load("train_embeddings.npy")

 import joblib, numpy as np, torch
 # Load model + embeddings
+model = ViTModel.from_pretrained("Sathya77/ViT_MvTec")
 knn = joblib.load("knn.pkl")
 train_embeddings = np.load("train_embeddings.npy")