version 1.0

README.md

@@ -1,8 +1,8 @@
 ---
-title: Snake Classification Using FastAI
-emoji: 📚
-colorFrom: indigo
-colorTo: green
+title: Basic Snake Classification
+emoji: 🔥
+colorFrom: dark-blue
+colorTo: light-pink
 sdk: gradio
 sdk_version: 3.24.1
 app_file: app.py
@@ -10,4 +10,10 @@ pinned: false
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+This is a basic snake classification model. 
+
+Built by fine-tuning the Levit 256 and using the "Identifying the different breeds of Snakes" (https://www.kaggle.com/datasets/duttadebadri/identifying-different-breeds-of-snakes) dataset on Kaggle.
+
+The notebook used for training the model can be found at: https://www.kaggle.com/code/utkmal/snake-breeds-classifier-using-fast-ai
+
+(Built by following the lessons 1 and 2 tutorial of FastAI)

app.py

@@ -0,0 +1,32 @@
+import gradio as gr
+from fastai.vision.all import *
+
+# Converting Unix Path to Windows Path
+import pathlib
+pathlib.PosixPath = pathlib.WindowsPath
+
+# Importing the model
+model = load_learner('model.pkl')
+
+categories = model.dls.vocab
+
+def predict_snake_type(img):
+    snake_type, idx, probs = model.predict(img)
+    return dict(zip(categories, map(float, probs)))
+
+# Gradio Interface
+input_img = gr.Image()
+outputs = gr.Label(num_top_classes=5)
+
+gr.Interface(
+    fn=predict_snake_type, 
+    inputs=[input_img], outputs=outputs,
+    examples=[
+        'examples/agkistrodon-contortrix.jpg', 
+        'examples/haldea-striatula.jpg', 
+        'examples/masticophis-flagellum.jpg',
+        'examples/storeria-occipitomaculata.jpg'
+        ],
+    title='Classifying Different Breeds of snakes',
+    description='This is a multi-classification model that identifies different breeds of snakes.\nPlease note that is model is only `74%` accurate.'
+).launch()

model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9a290f583373e743bb32ea8597b494c73fb34cd05063195f400d11f06aff0d3f
+size 78069633

requirements.txt

@@ -0,0 +1,2 @@
+fastai
+gradio