update app

app.ipynb

@@ -3,7 +3,7 @@
   {
    "cell_type": "code",
    "execution_count": 40,
-   "id": "d2b5c986",
+   "id": "c3638940",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -12,8 +12,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 41,
-   "id": "9f407c47",
+   "execution_count": 44,
+   "id": "79d0e887",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -28,7 +28,7 @@
   {
    "cell_type": "code",
    "execution_count": 4,
-   "id": "9063a1c9",
+   "id": "f4669ec7",
    "metadata": {},
    "outputs": [
     {
@@ -51,7 +51,7 @@
   {
    "cell_type": "code",
    "execution_count": 6,
-   "id": "4656eda3",
+   "id": "09cdb957",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -66,7 +66,7 @@
   {
    "cell_type": "code",
    "execution_count": 27,
-   "id": "82646005",
+   "id": "dad80553",
    "metadata": {
     "scrolled": true
    },
@@ -163,7 +163,7 @@
   {
    "cell_type": "code",
    "execution_count": 29,
-   "id": "77aa5f4e",
+   "id": "f0de21d7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -176,7 +176,7 @@
   {
    "cell_type": "code",
    "execution_count": 31,
-   "id": "bdd547c0",
+   "id": "4424abab",
    "metadata": {},
    "outputs": [
     {
@@ -321,7 +321,7 @@
   {
    "cell_type": "code",
    "execution_count": 34,
-   "id": "ad5ddfc9",
+   "id": "2bf0c23c",
    "metadata": {},
    "outputs": [
     {
@@ -388,7 +388,7 @@
   {
    "cell_type": "code",
    "execution_count": 42,
-   "id": "ac580214",
+   "id": "80d6199c",
    "metadata": {
     "scrolled": true
    },
@@ -407,10 +407,35 @@
     "print('Export successful')"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 49,
+   "id": "d5934911",
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "TypeError",
+     "evalue": "__init__() takes 2 positional arguments but 3 were given",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[1;31mTypeError\u001b[0m                                 Traceback (most recent call last)",
+      "\u001b[1;32m~\\AppData\\Local\\Temp\\ipykernel_10308\\3394884428.py\u001b[0m in \u001b[0;36m<module>\u001b[1;34m\u001b[0m\n\u001b[0;32m      1\u001b[0m \u001b[1;32mfrom\u001b[0m \u001b[0mfastai\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mvision\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mwidgets\u001b[0m \u001b[1;32mimport\u001b[0m \u001b[0mImageClassifierCleaner\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m      2\u001b[0m \u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m----> 3\u001b[1;33m \u001b[0mcleaner\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mImageClassifierCleaner\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mlearn\u001b[0m\u001b[1;33m,\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m\u001b[0;32m      4\u001b[0m \u001b[0mcleaner\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
+      "\u001b[1;31mTypeError\u001b[0m: __init__() takes 2 positional arguments but 3 were given"
+     ]
+    }
+   ],
+   "source": [
+    "from fastai.vision.widgets import ImageClassifierCleaner\n",
+    "\n",
+    "cleaner = ImageClassifierCleaner(learn)\n",
+    "cleaner"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "4f58c32f",
+   "id": "c88205d2",
    "metadata": {},
    "outputs": [],
    "source": []

app.py

@@ -5,7 +5,7 @@ learn = load_learner('model.pkl')
 
 def classify_img(img):
     pred,idx,probs = learn.predict(img)
-    return str(pred + "   Prob: " + str(round(torch.max(probs).item(), 4)) )
+    return str(pred + ",   probability: " + str(round(torch.max(probs).item(), 4)) )
 
 image = gr.inputs.Image(shape=(192, 192))
 label = gr.outputs.Label()