inital commit

model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83e5e041f686171106bb1365a2209670b153b321f905fea10fe8a910a086ab50
+size 46966605

monkey-classifer.py

@@ -0,0 +1,20 @@
+from fastai.vision.all import *
+import gradio as gr
+
+learn = load_learner("model.pkl")
+
+categories = ["Bald uakari monkey", "Mandrill monkey", 
+                "Panamanian White-faced Capuchin monkey",
+                "Golden snub-nosed monkey", 
+                "Black howler monkey", "Guinea baboon", 
+                "Bonnet macaque", "Japanese macaque"]
+
+def classify_image(image):
+    pred, i, prob = learn.predict(image)
+    return dict(zip(categories, map(float, prob)))
+
+image = gr.inputs.Image(shape=(192, 192))
+label = gr.outputs.Label()
+
+interface = gr.Interface(fn=classify_image, inputs=image, outputs=label)
+interface.launch(inline=False)

requirements.txt

@@ -0,0 +1,5 @@
+duckduckgo-search==2.8.5
+torch
+fastai
+notebook
+gradio