adding new embs

app.py

@@ -6,15 +6,15 @@ import wikipedia
 
 
 # Only runs once when the script is first run.
-with open("butts_1024_new.pickle", "rb") as handle:
+with open("insectarium_768.pickle", "rb") as handle:
     index = pickle.load(handle)
 
 # Load model for computing embeddings.
-feature_extractor = AutoFeatureExtractor.from_pretrained("sasha/autotrain-butterfly-similarity-2490576840")
-model = AutoModel.from_pretrained("sasha/autotrain-butterfly-similarity-2490576840")
+feature_extractor = AutoFeatureExtractor.from_pretrained("sasha/vit-base-butterflies")
+model = AutoModel.from_pretrained("sasha/vit-base-butterflies")
 
 # Candidate images.
-dataset = load_dataset("sasha/butterflies_10k_names_multiple")
+dataset = load_dataset("sasha/insectarium-butterflies")
 ds = dataset["train"]
 
 
@@ -27,45 +27,46 @@ def query(image, top_k=4):
     logits = results[1][0].tolist()
     images = ds.select(inx)["image"]
     captions = ds.select(inx)["name"]
-    images_with_captions = [(i, c) for i, c in zip(images,captions)]
-    labels_with_probs = dict(zip(captions,logits))
-    labels_with_probs = {k: 1- v for k, v in labels_with_probs.items()}
+    images_with_captions = [(i, c) for i, c in zip(images, captions)]
+    labels_with_probs = dict(zip(captions, logits))
+    labels_with_probs = {k: 1 - v for k, v in labels_with_probs.items()}
     try:
-    	description = wikipedia.summary(captions[0], sentences = 1)
-    	description = "### " + description
-    	url = wikipedia.page(captions[0]).url
-    	url = " You can learn more about your butterfly [here](" + str(url) + ")!"
-    	description = description + url
+        description = wikipedia.summary(captions[0], sentences=1)
+        description = "### " + description
+        url = wikipedia.page(captions[0]).url
+        url = " You can learn more about your butterfly [here](" + str(url) + ")!"
+        description = description + url
     except:
-    	description = "### Butterflies are insects in the order Lepidoptera, which also includes moths. Adult butterflies have large, often brightly coloured wings."
-    	url = "https://en.wikipedia.org/wiki/Butterfly"
-    	url = " You can learn more about butterflies [here](" + str(url) + ")!"
-    	description = description + url
+        description = "### Butterflies are insects in the order Lepidoptera, which also includes moths. Adult butterflies have large, often brightly coloured wings."
+        url = "https://en.wikipedia.org/wiki/Butterfly"
+        url = " You can learn more about butterflies [here](" + str(url) + ")!"
+        description = description + url
     return images_with_captions, labels_with_probs, description
 
 
 with gr.Blocks() as demo:
-	gr.Markdown("# Find my Butterfly 🦋")
-	gr.Markdown("## Use this Space to find your butterfly, based on the [iNaturalist butterfly dataset](https://huggingface.co/datasets/huggan/inat_butterflies_top10k)!")
-	with gr.Row():
-		with gr.Column(scale=1):
-			inputs = gr.Image()
-			btn = gr.Button("Find my butterfly!")
-			description = gr.Markdown()
-			
-		with gr.Column(scale=2):
-			outputs=gr.Gallery(rows=2)
-			labels = gr.Label()
-			
-	gr.Markdown("### Image Examples")
-	gr.Examples(
-	examples=["elton.jpg", "ken.jpg", "gaga.jpg", "taylor.jpg"],
-	inputs=inputs,
-	outputs=[outputs,labels],
-	fn=query,
-	cache_examples=True,
-	)
-	btn.click(query, inputs, [outputs, labels, description])
-	
+    gr.Markdown("# Find my Butterfly 🦋")
+    gr.Markdown(
+        "## Use this Space to find your butterfly, based on the [iNaturalist butterfly dataset](https://huggingface.co/datasets/huggan/inat_butterflies_top10k)!"
+    )
+    with gr.Row():
+        with gr.Column(scale=1):
+            inputs = gr.Image()
+            btn = gr.Button("Find my butterfly!")
+            description = gr.Markdown()
+
+        with gr.Column(scale=2):
+            outputs = gr.Gallery(rows=2)
+            labels = gr.Label()
+
+    gr.Markdown("### Image Examples")
+    gr.Examples(
+        examples=["elton.jpg", "ken.jpg", "gaga.jpg", "taylor.jpg"],
+        inputs=inputs,
+        outputs=[outputs, labels],
+        fn=query,
+        cache_examples=True,
+    )
+    btn.click(query, inputs, [outputs, labels, description])
+
 demo.launch()
-    

index_768_cosine.pickle

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:864fe29de71f0e5b56ca87b04d559ea707d4cd3798429f80f04b8a58a07e3721
-size 53168791

butts_1024_new.pickle → insectarium_768.pickle

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d5150f73211c6bac942b724c0bd56320ad99a0a0142d2c42199cab6bf719a1db
-size 52935099
+oid sha256:d8ceee3f2f9388816bc6d37101be7cabe6b62cf140729bd9d13e4386e3265791
+size 211226535