Minor bug fix in extracting the predicted superhero

app.ipynb

@@ -171,12 +171,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 32,
    "id": "12b3519f",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T04:18:14.466326Z",
-     "start_time": "2023-06-04T04:18:14.460213Z"
+     "end_time": "2023-06-04T04:27:08.756981Z",
+     "start_time": "2023-06-04T04:27:08.748813Z"
     }
    },
    "outputs": [
@@ -198,10 +198,10 @@
     "    pred, idx, probs = learn.predict(img)\n",
     "    \n",
     "    prediction = dict(zip(categories, map(float, probs)))\n",
-    "    #print (f\"prediction = {prediction}\")\n",
+    "    print (f\"prediction = {prediction}\")\n",
     "    \n",
-    "    predicted_hero = categories[np.argsort(prediction.values())[0]]\n",
-    "    #print (f\"predicted_hero = {predicted_hero}\")\n",
+    "    predicted_hero = max(prediction, key=lambda key: prediction[key])\n",
+    "    print (f\"predicted_hero = {predicted_hero}\")\n",
     "    \n",
     "    if predicted_hero == 'Superman':\n",
     "        alter_ego = \"Clark Kent Jr\"\n",
@@ -217,12 +217,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 33,
    "id": "0be42346",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T04:18:15.324202Z",
-     "start_time": "2023-06-04T04:18:15.281352Z"
+     "end_time": "2023-06-04T04:27:08.982499Z",
+     "start_time": "2023-06-04T04:27:08.904473Z"
     }
    },
    "outputs": [
@@ -263,6 +263,14 @@
      "metadata": {},
      "output_type": "display_data"
     },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "prediction = {'Batman': 0.940523087978363, 'Flash': 0.00284947850741446, 'Superman': 0.05662744119763374}\n",
+      "predicted_hero = Batman\n"
+     ]
+    },
     {
      "data": {
       "text/plain": [
@@ -272,7 +280,7 @@
        " 'Bruce Wayne')"
       ]
      },
-     "execution_count": 6,
+     "execution_count": 33,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -283,12 +291,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
-   "id": "826388a1",
+   "execution_count": 34,
+   "id": "ad2873f2",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T04:18:16.173080Z",
-     "start_time": "2023-06-04T04:18:15.829127Z"
+     "end_time": "2023-06-04T04:27:09.939658Z",
+     "start_time": "2023-06-04T04:27:09.574763Z"
     }
    },
    "outputs": [
@@ -308,12 +316,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 35,
    "id": "d845ec5f",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T04:18:18.233242Z",
-     "start_time": "2023-06-04T04:18:16.393046Z"
+     "end_time": "2023-06-04T04:27:11.072944Z",
+     "start_time": "2023-06-04T04:27:09.942171Z"
     }
    },
    "outputs": [
@@ -321,7 +329,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Running on local URL:  http://127.0.0.1:7860\n",
+      "Running on local URL:  http://127.0.0.1:7864\n",
       "\n",
       "To create a public link, set `share=True` in `launch()`.\n"
      ]
@@ -329,7 +337,7 @@
     {
      "data": {
       "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7860/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
+       "<div><iframe src=\"http://127.0.0.1:7864/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
       ],
       "text/plain": [
        "<IPython.core.display.HTML object>"
@@ -342,7 +350,7 @@
      "data": {
       "text/plain": []
      },
-     "execution_count": 8,
+     "execution_count": 35,
      "metadata": {},
      "output_type": "execute_result"
     },
@@ -383,6 +391,104 @@
      "metadata": {},
      "output_type": "display_data"
     },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "prediction = {'Batman': 5.742020903198863e-07, 'Flash': 3.5911152735934593e-06, 'Superman': 0.9999958276748657}\n",
+      "predicted_hero = Superman\n"
+     ]
+    },
+    {
+     "data": {
+      "text/html": [
+       "\n",
+       "<style>\n",
+       "    /* Turns off some styling */\n",
+       "    progress {\n",
+       "        /* gets rid of default border in Firefox and Opera. */\n",
+       "        border: none;\n",
+       "        /* Needs to be in here for Safari polyfill so background images work as expected. */\n",
+       "        background-size: auto;\n",
+       "    }\n",
+       "    progress:not([value]), progress:not([value])::-webkit-progress-bar {\n",
+       "        background: repeating-linear-gradient(45deg, #7e7e7e, #7e7e7e 10px, #5c5c5c 10px, #5c5c5c 20px);\n",
+       "    }\n",
+       "    .progress-bar-interrupted, .progress-bar-interrupted::-webkit-progress-bar {\n",
+       "        background: #F44336;\n",
+       "    }\n",
+       "</style>\n"
+      ],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "prediction = {'Batman': 2.2285334125626832e-05, 'Flash': 0.9998688697814941, 'Superman': 0.00010885472147492692}\n",
+      "predicted_hero = Flash\n"
+     ]
+    },
+    {
+     "data": {
+      "text/html": [
+       "\n",
+       "<style>\n",
+       "    /* Turns off some styling */\n",
+       "    progress {\n",
+       "        /* gets rid of default border in Firefox and Opera. */\n",
+       "        border: none;\n",
+       "        /* Needs to be in here for Safari polyfill so background images work as expected. */\n",
+       "        background-size: auto;\n",
+       "    }\n",
+       "    progress:not([value]), progress:not([value])::-webkit-progress-bar {\n",
+       "        background: repeating-linear-gradient(45deg, #7e7e7e, #7e7e7e 10px, #5c5c5c 10px, #5c5c5c 20px);\n",
+       "    }\n",
+       "    .progress-bar-interrupted, .progress-bar-interrupted::-webkit-progress-bar {\n",
+       "        background: #F44336;\n",
+       "    }\n",
+       "</style>\n"
+      ],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "prediction = {'Batman': 0.016343194991350174, 'Flash': 0.002146175829693675, 'Superman': 0.981510579586029}\n",
+      "predicted_hero = Superman\n"
+     ]
+    },
     {
      "data": {
       "text/html": [
@@ -419,6 +525,14 @@
      },
      "metadata": {},
      "output_type": "display_data"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "prediction = {'Batman': 0.9863764643669128, 'Flash': 0.0022053574211895466, 'Superman': 0.011418282985687256}\n",
+      "predicted_hero = Batman\n"
+     ]
     }
    ],
    "source": [
@@ -450,12 +564,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 36,
    "id": "b942385f",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T04:18:28.883975Z",
-     "start_time": "2023-06-04T04:18:28.532380Z"
+     "end_time": "2023-06-04T04:27:21.444358Z",
+     "start_time": "2023-06-04T04:27:21.433263Z"
     }
    },
    "outputs": [

app.py

@@ -22,10 +22,10 @@ def classify_image(img):
     pred, idx, probs = learn.predict(img)
     
     prediction = dict(zip(categories, map(float, probs)))
-    #print (f"prediction = {prediction}")
+    print (f"prediction = {prediction}")
     
-    predicted_hero = categories[np.argsort(prediction.values())[0]]
-    #print (f"predicted_hero = {predicted_hero}")
+    predicted_hero = max(prediction, key=lambda key: prediction[key])
+    print (f"predicted_hero = {predicted_hero}")
     
     if predicted_hero == 'Superman':
         alter_ego = "Clark Kent Jr"