initial commit

app.py

@@ -0,0 +1,13 @@
+from fastai.vision.all import *
+import gradio as gr
+
+genres  = ('rock, jazz')
+learner = load_learner('model.pkl')
+
+def predict_genre(img):
+    pred, idx, probs = learner.predict(img)
+    return dict(zip(genres, map(float, probs)))
+
+
+demo = gr.Interface(fn=predict_genre, inputs="image", outputs="label")
+demo.launch()

model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:13fcaada4f339f7e502f441f590de9c0bf79beff32f9f54ad701c0e42f38720e
+size 46962347