updated demo to include 50 classes

app.py

@@ -1,53 +1,72 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: app.ipynb.
 
 # %% auto 0
-__all__ = ['repo_id', 'learner', 'path', 'categories', 'title', 'description', 'article', 'image', 'label', 'examples', 'intf',
-           'classify_image']
+__all__ = ['repo_id', 'learner', 'path', 'countries', 'categories', 'title', 'description', 'article', 'image', 'label',
+           'country', 'summary', 'link', 'examples', 'intf', 'get_countries', 'classify_image']
 
-# %% app.ipynb 2
+# %% app.ipynb 3
 from fastai.vision.all import *
 from huggingface_hub import from_pretrained_fastai
 import gradio as gr
+import wikipedia
+import pandas as pd
 
-# %% app.ipynb 3
-repo_id = "Jimmie/snake-image-classification"
+# %% app.ipynb 4
+repo_id = "Jimmie/snake-species-identification"
 
 # loading the model from huggingface_hub
 learner = from_pretrained_fastai(repo_id)
 
-# %% app.ipynb 4
+# %% app.ipynb 5
 path = Path('demo-images/')
-
-# %% app.ipynb 14
+countries = pd.read_csv('species_to_country_mapping.csv', index_col=0)
+
+# %% app.ipynb 9
+def get_countries(binomial):
+    sample_row = countries.loc[binomial]
+    country_list = sample_row[sample_row == 1].index.tolist()
+    # title case all items in country_list
+    country_list = [country.title() for country in country_list]
+    # return all items in country_list as a string
+    return ", ".join(country_list)
+
+# %% app.ipynb 20
 categories = tuple(learner.dls.vocab)
 
 def classify_image(img):
     pred,idx,probs = learner.predict(img)
-    return dict(zip(categories, map(float, probs)))
+    countries = get_countries(pred)
+    summary = wikipedia.summary(pred)
+    wiki_link = f"Learn more: <a href={wikipedia.page(pred).url}>{pred}</a>"
+    return dict(zip(categories, map(float, probs))), countries, summary, wiki_link
 
-# %% app.ipynb 16
-title = "Snake Image Classification"
+# %% app.ipynb 22
+title = "Snake Species Identification"
 
 description = """
-This demo is an ongoing iteration of a [bigger project](https://github.com/jimmiemunyi/the-snake-project) meant to classify snakes as venomous or non-venomous.
+This demo is an ongoing iteration of the [Snake Species Identification](https://github.com/jimmiemunyi/the-snake-project-cls) project meant to classify snakes up to the species level (binomial name).
 
-Currently, it can classify snakes into 10 genera.
+Currently, it can classify snakes into 50 categories but it is continually updated to support more categories (over 200).
 
-The model can be found here: https://huggingface.co/Jimmie/snake-image-classification
- 
+The model can be found here: https://huggingface.co/Jimmie/snake-species-identification.
+The model is trained on the following dataset: https://www.aicrowd.com/challenges/snakeclef2021-snake-species-identification-challenge.
 
 Enjoy!
 """
 
-article = "Blog posts on how the model is being trained: ."
+article = "Blog posts on how the model is being trained: COMING SOON!"
 
 
-image = gr.inputs.Image(shape=(224, 224))
-label = gr.outputs.Label()
+image = gr.Image(shape=(224, 224))
+label = gr.Label(num_top_classes=3, label='Binomial')
+country = gr.Textbox(label='Countries where the species is found')
+summary = gr.Textbox(label='Wikipedia Summary')
+link = gr.HTML(label="Learn More:", show_label=True)
 examples = list(path.ls())
 
 
-intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples,
-                   title = title, description = description, article = article,
-                   enable_queue=True, cache_examples=False)
+intf = gr.Interface(fn=classify_image, inputs=image, 
+                   outputs=[label, country, summary, link], examples=examples,
+                   title = title, description = description, article = article, 
+                   cache_examples=False)
 intf.launch(inline=False)

flagged/Binomial/tmphs9nl929.json

@@ -0,0 +1 @@
+{"label": "Crotalus horridus", "confidences": [{"label": "Crotalus horridus", "confidence": 0.999326229095459}, {"label": "Pantherophis spiloides", "confidence": 0.00014710254617966712}, {"label": "Pantherophis alleghaniensis", "confidence": 5.688565943273716e-05}]}

flagged/log.csv

@@ -0,0 +1,2 @@
+img,Binomial,Summary,flag,username,timestamp
+/home/eleven/code/personal/snake-image-classification/flagged/img/tmp0o3vkpoo.jpg,/home/eleven/code/personal/snake-image-classification/flagged/Binomial/tmphs9nl929.json,"The timber rattlesnake, canebrake rattlesnake, or banded rattlesnake (Crotalus horridus) is a species of pit viper endemic to eastern North America. Like all other pit vipers, it is venomous, with a very toxic bite. C. horridus is the only rattlesnake species in most of the populous Northeastern United States and is second only to its relatives to the west, the prairie rattlesnake, as the most northerly distributed venomous snake in North America. No subspecies are currently recognized.",,,2023-03-27 07:54:18.071444

requirements.txt

@@ -1,3 +1,6 @@
+wikipedia==1.4.0
+pandas==1.5.2
+toml==0.10.2
 fastai<=2.7.11
 timm==0.8.3dev0
-git+https://github.com/jimmiemunyi/the-snake-project@main#egg=the-snake-project
+git+https://github.com/jimmiemunyi/the-snake-project@main#egg=the-snake-project