adding cache and wikipedia

app.py

@@ -2,7 +2,7 @@ import pickle
 import gradio as gr
 from datasets import load_dataset
 from transformers import AutoModel, AutoFeatureExtractor
-
+import wikipedia
 
 # Only runs once when the script is first run.
 with open("slugs_index_1024_cosine.pickle", "rb") as handle:
@@ -29,19 +29,34 @@ def query(image, top_k=4):
     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()}
-    return images_with_captions, labels_with_probs
-
+    try:
+    	description = wikipedia.summary(captions[0], sentences = 1)
+    	description = "## " + description
+    	url = wikipedia.page(captions[0]).url
+    	url = " You can learn more about your slug [here](" + str(url) + ")!"
+    	description = description + url
+    except:
+    	description = "## Sea slugs, or Nudibranchs, are marine invertebrates that often live in reefs underwater. They have an enormous variation in body shape, color, and size."
+    	url = "https://en.wikipedia.org/wiki/Sea_slug"
+    	url = " You can learn more about your slug [here](" + str(url) + ")!"
+    	description = description + url
+    return images_with_captions, labels_with_probs, description
+    
+    
 
 with gr.Blocks() as demo:
-	gr.Markdown("# Find my Sea Slug 🐌")
+	gr.Markdown("# Which Slug Am I ? 🐌")
 	gr.Markdown("## Use this Space to find your sea slug, based on the [Nudibranchs of the Sunshine Coast Australia dataset](https://huggingface.co/datasets/sasha/australian_sea_slugs)!")
 	with gr.Row():
 		with gr.Column(min_width= 900):
 			inputs = gr.Image(shape=(800, 1600))
 			btn = gr.Button("Find my sea slug 🐌!")
+			description = gr.Markdown()
+			
 		with gr.Column():
 			outputs=gr.Gallery().style(grid=[2], height="auto")
 			labels = gr.Label()
+			
 	gr.Markdown("### Image Examples")
 	gr.Examples(
 	examples=["elton.jpg", "ken.jpg", "gaga.jpg", "taylor.jpg"],
@@ -50,7 +65,7 @@ with gr.Blocks() as demo:
 	fn=query,
 	cache_examples=True,
 	)
-	btn.click(query, inputs, [outputs, labels])
+	btn.click(query, inputs, [outputs, labels, description])
 	
 demo.launch()
     

gradio_cached_examples/11/component 0/8d270ec8-da00-471b-a1ae-e78efae812d4/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/11/component 0/8d270ec8-da00-471b-a1ae-e78efae812d4/tmpvt7_pclkqu7ayp6_.png": "Philinopsis pilsbryi", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/11/component 0/8d270ec8-da00-471b-a1ae-e78efae812d4/tmpi67ptxcr1aikli53.png": "Philinopsis pilsbryi", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/11/component 0/8d270ec8-da00-471b-a1ae-e78efae812d4/tmpyv0h2ynmpd2jpicg.png": "Halgerda aurantiomaculata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/11/component 0/8d270ec8-da00-471b-a1ae-e78efae812d4/tmp7dcr9lz8l_aoi7q0.png": "Micromelo undata"}

gradio_cached_examples/11/component 1/tmp85up16jz.json

@@ -0,0 +1 @@
+{"label": "Philinopsis pilsbryi", "confidences": [{"label": "Philinopsis pilsbryi", "confidence": 0.0}, {"label": "Halgerda aurantiomaculata", "confidence": 0.0}, {"label": "Micromelo undata", "confidence": 0.0}]}

gradio_cached_examples/11/log.csv

@@ -0,0 +1,2 @@
+component 0,component 1,flag,username,timestamp
+/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/11/component 0/8d270ec8-da00-471b-a1ae-e78efae812d4,/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/11/component 1/tmp85up16jz.json,,,2022-12-16 13:18:08.434686

gradio_cached_examples/12/component 0/51ba82b5-3cff-45ff-b32d-76dfd7f4ce0e/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/51ba82b5-3cff-45ff-b32d-76dfd7f4ce0e/tmpakfa_y7tp6c0chz9.png": "Pleurobranchus peronii", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/51ba82b5-3cff-45ff-b32d-76dfd7f4ce0e/tmpd3xmsincip5hv64y.png": "Pleurobranchus peronii", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/51ba82b5-3cff-45ff-b32d-76dfd7f4ce0e/tmpin3b_0zals1_verk.png": "Sclerodoris cf. coriacea", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/51ba82b5-3cff-45ff-b32d-76dfd7f4ce0e/tmpfi_vqg61s87semoo.png": "Sclerodoris cf. coriacea"}

gradio_cached_examples/12/component 0/883dcfc1-7614-4fb7-bf38-0668cffb6ea1/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/883dcfc1-7614-4fb7-bf38-0668cffb6ea1/tmpf2q2zcbtiip4rnuc.png": "Fiona pinnata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/883dcfc1-7614-4fb7-bf38-0668cffb6ea1/tmppz_669fg4b66b2uo.png": "Phestilla melanobrachia", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/883dcfc1-7614-4fb7-bf38-0668cffb6ea1/tmp80hd4x358o1hb8gp.png": "Atagema albata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/883dcfc1-7614-4fb7-bf38-0668cffb6ea1/tmpw1n3jhgyt4pc3j8j.png": "Phestilla minor"}

gradio_cached_examples/12/component 0/aa88d0a4-3a2e-4848-9e7c-c8268b93da9d/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/aa88d0a4-3a2e-4848-9e7c-c8268b93da9d/tmpdxnbb813gk3i3ja9.png": "Hallaxa indecora", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/aa88d0a4-3a2e-4848-9e7c-c8268b93da9d/tmpg5jhrmcwzl_gngc3.png": "Rostanga bifurcata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/aa88d0a4-3a2e-4848-9e7c-c8268b93da9d/tmp3eejeykpdr6r4y9k.png": "(Crosse & Fischer, 1865)", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/aa88d0a4-3a2e-4848-9e7c-c8268b93da9d/tmpuw66_be05hqh9rqk.png": "Pleurobranchus caledonicus"}

gradio_cached_examples/12/component 0/bb8a3397-4226-4bdf-a706-7e38e0c65c3a/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/bb8a3397-4226-4bdf-a706-7e38e0c65c3a/tmpuaydr6ocotehg_dj.png": "Dendrodoris nigra", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/bb8a3397-4226-4bdf-a706-7e38e0c65c3a/tmpoiy7whrounxx0xw_.png": "Chelidonura inornata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/bb8a3397-4226-4bdf-a706-7e38e0c65c3a/tmp49nmc9ivhjpmtfuj.png": "Chelidonura inornata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/bb8a3397-4226-4bdf-a706-7e38e0c65c3a/tmpzxy5_sga1qhwhcxh.png": "Chelidonura inornata"}

gradio_cached_examples/12/component 1/tmp8wigsoc3.json

@@ -0,0 +1 @@
+{"label": "Dendrodoris nigra", "confidences": [{"label": "Dendrodoris nigra", "confidence": 0.19950909942147177}, {"label": "Chelidonura inornata", "confidence": 0.15343128095999126}]}

gradio_cached_examples/12/component 1/tmpg4e9ih25.json

@@ -0,0 +1 @@
+{"label": "Hallaxa indecora", "confidences": [{"label": "Hallaxa indecora", "confidence": 0.15913860581831485}, {"label": "Rostanga bifurcata", "confidence": 0.15435959463431237}, {"label": "(Crosse & Fischer, 1865)", "confidence": 0.15100633689905418}, {"label": "Pleurobranchus caledonicus", "confidence": 0.1485400156066694}]}

gradio_cached_examples/12/component 1/tmph956q98z.json

@@ -0,0 +1 @@
+{"label": "Fiona pinnata", "confidences": [{"label": "Fiona pinnata", "confidence": 0.1405049929835933}, {"label": "Phestilla melanobrachia", "confidence": 0.12569111690849366}, {"label": "Atagema albata", "confidence": 0.12517943485935834}, {"label": "Phestilla minor", "confidence": 0.12276734601804518}]}

gradio_cached_examples/12/component 1/tmpwfcniknp.json

@@ -0,0 +1 @@
+{"label": "Pleurobranchus peronii", "confidences": [{"label": "Pleurobranchus peronii", "confidence": 0.19036074978457096}, {"label": "Sclerodoris cf. coriacea", "confidence": 0.17929956787471024}]}

gradio_cached_examples/12/log.csv

@@ -0,0 +1,5 @@
+component 0,component 1,flag,username,timestamp
+/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/51ba82b5-3cff-45ff-b32d-76dfd7f4ce0e,/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 1/tmpwfcniknp.json,,,2022-12-16 13:38:37.058556
+/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/883dcfc1-7614-4fb7-bf38-0668cffb6ea1,/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 1/tmph956q98z.json,,,2022-12-16 13:38:38.312889
+/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/aa88d0a4-3a2e-4848-9e7c-c8268b93da9d,/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 1/tmpg4e9ih25.json,,,2022-12-16 13:38:39.598397
+/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 0/bb8a3397-4226-4bdf-a706-7e38e0c65c3a,/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/12/component 1/tmp8wigsoc3.json,,,2022-12-16 13:38:40.984197

gradio_cached_examples/13/component 0/04d95beb-9b2f-420a-b5b7-66ec14f3ad63/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/04d95beb-9b2f-420a-b5b7-66ec14f3ad63/tmp92f7ql55xlpj7ua5.png": "Fiona pinnata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/04d95beb-9b2f-420a-b5b7-66ec14f3ad63/tmpmxml9muugdo_b8ql.png": "Phestilla melanobrachia", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/04d95beb-9b2f-420a-b5b7-66ec14f3ad63/tmpquwxclf917j89k2s.png": "Atagema albata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/04d95beb-9b2f-420a-b5b7-66ec14f3ad63/tmpe87lsvks8rlcrg4k.png": "Phestilla minor"}

gradio_cached_examples/13/component 0/26b50a07-a95e-422e-a55c-96c8cbeae7b3/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/26b50a07-a95e-422e-a55c-96c8cbeae7b3/tmp58nfl7apx8t1gj2r.png": "Hallaxa indecora", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/26b50a07-a95e-422e-a55c-96c8cbeae7b3/tmpmlxbajbeqjnxs7zx.png": "Rostanga bifurcata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/26b50a07-a95e-422e-a55c-96c8cbeae7b3/tmp7jo7zez_7o6xi7z6.png": "(Crosse & Fischer, 1865)", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/26b50a07-a95e-422e-a55c-96c8cbeae7b3/tmp0061410gjx5ron08.png": "Pleurobranchus caledonicus"}

gradio_cached_examples/13/component 0/8f453916-2755-44a6-92ff-2b74d75c2eb3/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/8f453916-2755-44a6-92ff-2b74d75c2eb3/tmpww1e33ikf49_v_d0.png": "Dendrodoris nigra", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/8f453916-2755-44a6-92ff-2b74d75c2eb3/tmpwv_nxzg1s3h2vew1.png": "Chelidonura inornata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/8f453916-2755-44a6-92ff-2b74d75c2eb3/tmpo7_s4yu19te1inkd.png": "Chelidonura inornata", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/8f453916-2755-44a6-92ff-2b74d75c2eb3/tmpmdmlgk6xhmx5b2xo.png": "Chelidonura inornata"}

gradio_cached_examples/13/component 0/a7809c84-4636-43a4-a3f3-1233032ddca7/captions.json

@@ -0,0 +1 @@
+{"/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/a7809c84-4636-43a4-a3f3-1233032ddca7/tmp30yec4khshq2pcbm.png": "Pleurobranchus peronii", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/a7809c84-4636-43a4-a3f3-1233032ddca7/tmprtc8ksijq2s3ezz1.png": "Pleurobranchus peronii", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/a7809c84-4636-43a4-a3f3-1233032ddca7/tmpo7rehi0hpluiexrx.png": "Sclerodoris cf. coriacea", "/home/sasha/Documents/HuggingFace/find-my-sea-slug/gradio_cached_examples/13/component 0/a7809c84-4636-43a4-a3f3-1233032ddca7/tmp3tko5jt0rp5y11us.png": "Sclerodoris cf. coriacea"}