Initial Commit

app.py

@@ -0,0 +1,34 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: app.ipynb.
+
+# %% auto 0
+__all__ = ['learn', 'labels', 'examples', 'add_image_label', 'predict']
+
+# %% app.ipynb 2
+import timm
+from fastai.vision.all import *
+import gradio as gr
+
+
+def add_image_label(x):
+    names = ["gond", "kalighat", "kangra", "madhubani", "mandana", "pichwai", "warli"]
+    for name in names:
+        if name in Path(x).name:
+            return name
+        elif "kerala" in Path(x).name:
+            return "kerala mural"
+    return "unknown"
+
+# %% app.ipynb 4
+learn = load_learner("model.pkl")
+
+# %% app.ipynb 6
+labels = learn.dls.vocab
+def predict(img):
+    img = PILImage.create(img)
+    pred,pred_idx,probs = learn.predict(img)
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+
+
+# %% app.ipynb 8
+examples = ["examples//warli.jpeg", "examples//madhubani.jpg"]
+gr.Interface(fn=predict, inputs=gr.Image(type="pil"), outputs=gr.Label(num_top_classes=3), examples = examples, title = "Indian Art Classifier").launch(inline=False, debug=True)

model.pkl

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

requirements.txt

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