Update similarity_utils.py

similarity_utils.py

@@ -18,7 +18,7 @@ np.random.seed(seed)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load model for computing embeddings..
-model_ckpt = "nateraw/vit-base-beans"
+model_ckpt = "yangswei/snacks_classification"
 extractor = AutoFeatureExtractor.from_pretrained(model_ckpt)
 
 # Data transformation chain.