Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,29 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: 02a_prod_artsie.ipynb.
+
+# %% auto 0
+__all__ = ['repo_model_name', 'remote_learner', 'labels', 'demo', 'classify_img']
+
+# %% 02a_prod_artsie.ipynb 4
+from fastai.vision.all import *
+from huggingface_hub import push_to_hub_fastai, from_pretrained_fastai
+import gradio as gr
+
+# %% 02a_prod_artsie.ipynb 37
+repo_model_name = 'artsie-model'
+
+# %% 02a_prod_artsie.ipynb 40
+remote_learner = from_pretrained_fastai(f"rahuketu86/{repo_model_name}"); remote_learner
+
+# %% 02a_prod_artsie.ipynb 47
+labels = remote_learner.dls.vocab
+
+def classify_img(img):
+    img = PILImage.create(img)
+    pred, pred_idx, probs = remote_learner.predict(img)
+    return {labels[i]:float(probs[i]) for i in range(len(labels))}
+
+# %% 02a_prod_artsie.ipynb 50
+demo = gr.Interface(fn=classify_img, 
+                    inputs=gr.Image(), 
+                    outputs=gr.Label(num_top_classes=len(labels)))
+demo.launch(server_name="0.0.0.0", share=True)