Add missing function classify_image

.ipynb_checkpoints/app-checkpoint.ipynb

@@ -573,6 +573,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "#|export\n",
     "categories = ('Dog', 'Cat')\n",
     "def classify_image(img):\n",
     "    pref,idx,probs = learn.predict(img)\n",
@@ -716,13 +717,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 76,
+   "execution_count": 77,
    "id": "5e35eaab",
    "metadata": {},
    "outputs": [],
    "source": [
     "nbdev.export.nb_export('app.ipynb', '.')"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7f0bd0f2",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {

app.ipynb

@@ -573,6 +573,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "#|export\n",
     "categories = ('Dog', 'Cat')\n",
     "def classify_image(img):\n",
     "    pref,idx,probs = learn.predict(img)\n",
@@ -727,7 +728,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9480715a",
+   "id": "7f0bd0f2",
    "metadata": {},
    "outputs": [],
    "source": []

app.py

@@ -1,7 +1,7 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: app.ipynb.
 
 # %% auto 0
-__all__ = ['learn', 'image', 'label', 'examples', 'intf', 'is_cat']
+__all__ = ['learn', 'categories', 'image', 'label', 'examples', 'intf', 'is_cat', 'classify_image']
 
 # %% app.ipynb 2
 import fastbook
@@ -21,6 +21,12 @@ def is_cat(x): return x[0].issuper()
 # %% app.ipynb 11
 learn = load_learner('model.pkl')
 
+# %% app.ipynb 13
+categories = ('Dog', 'Cat')
+def classify_image(img):
+    pref,idx,probs = learn.predict(img)
+    return dict(zip(categories, map(float, probs)))
+
 # %% app.ipynb 16
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label()