deploy #1

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+minst.pk1 filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,6 +1,6 @@
 ---
 title: Musical Instrument Classifier
-emoji: 🌖
+emoji: 🎹
 colorFrom: indigo
 colorTo: gray
 sdk: gradio

app.py

@@ -0,0 +1,20 @@
+
+import gradio as gr
+from fastai.vision.all import *
+
+def classify_img(img):
+    minst,_,prob = learn.predict(PILImage.create(img))
+    return dict(zip(categories, map(float, prob)))
+
+#paths
+mdl = 'minst.pk1'
+samples = 'examples/'
+
+# load pre-trained model 
+learn = load_learner(mdl)
+
+categories = ('acoustic guitar', 'drums', 'Electric guitar', 'keyboard')
+examples = [f'{samples}acoustic guitar.jpg', f'{samples}drums.jpg', f'{samples}electric guitar.jpg', f'{samples}keyboard.jpg']
+
+demo = gr.Interface(fn=classify_img, inputs="image", outputs="label", examples=examples)
+demo.launch()

minst.pk1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:52767d004dd162a9fc9f54b64bf05ef63147f8c04316f97c0ba13be47d922ac3
+size 102874606