Update app.py

app.py

@@ -20,7 +20,8 @@ with open("lsh.pickle", "rb") as handle:
     loaded_lsh = pickle.load(handle)
 
 # Load model for computing embeddings.
-model_ckpt = "google/vit-base-patch16-224"
+# model_ckpt = "google/vit-base-patch16-224"
+model_ckpt = "anggtpd/snacks_classifier"
 model = AutoModel.from_pretrained(model_ckpt)
 lsh_builder = BuildLSHTable(model)
 lsh_builder.lsh = loaded_lsh