Update app.py

app.py

@@ -1,6 +1,4 @@
-"""
-Thanks to Freddy Boulton (https://github.com/freddyaboulton) for helping with this.
-"""
+
 
 
 import pickle
@@ -20,8 +18,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 = "matteopilotto/vit-base-patch16-224-in21k-snacks"
+model_ckpt = "google/vit-base-patch16-224"
+#model_ckpt = "matteopilotto/vit-base-patch16-224-in21k-snacks"
 model = AutoModel.from_pretrained(model_ckpt)
 lsh_builder = BuildLSHTable(model)
 lsh_builder.lsh = loaded_lsh