Minor bug fix in app

README.md

@@ -1,6 +1,6 @@
 ---
 title: Who Is The Hero
-emoji: 💻
+emoji: 🦸
 colorFrom: blue
 colorTo: yellow
 sdk: gradio
@@ -9,5 +9,3 @@ app_file: app.py
 pinned: false
 license: apache-2.0
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.ipynb

@@ -25,12 +25,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 51,
+   "execution_count": 1,
    "id": "754b56c0",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:54:49.353855Z",
-     "start_time": "2023-06-04T03:54:49.348893Z"
+     "end_time": "2023-06-04T04:18:12.302048Z",
+     "start_time": "2023-06-04T04:18:10.013098Z"
     }
    },
    "outputs": [],
@@ -45,12 +45,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 52,
+   "execution_count": 2,
    "id": "e4be8667",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:54:49.582532Z",
-     "start_time": "2023-06-04T03:54:49.489341Z"
+     "end_time": "2023-06-04T04:18:13.041741Z",
+     "start_time": "2023-06-04T04:18:12.948260Z"
     },
     "scrolled": true
    },
@@ -69,7 +69,7 @@
        "PILImage mode=RGB size=148x192"
       ]
      },
-     "execution_count": 52,
+     "execution_count": 2,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -83,12 +83,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 53,
+   "execution_count": 3,
    "id": "c7765b36",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:54:50.175615Z",
-     "start_time": "2023-06-04T03:54:50.113295Z"
+     "end_time": "2023-06-04T04:18:13.513099Z",
+     "start_time": "2023-06-04T04:18:13.479749Z"
     }
    },
    "outputs": [],
@@ -99,12 +99,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 54,
+   "execution_count": 4,
    "id": "92e8d6b4",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:54:50.331722Z",
-     "start_time": "2023-06-04T03:54:50.283858Z"
+     "end_time": "2023-06-04T04:18:13.811788Z",
+     "start_time": "2023-06-04T04:18:13.746509Z"
     }
    },
    "outputs": [
@@ -149,8 +149,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "CPU times: user 50.7 ms, sys: 40.4 ms, total: 91.2 ms\n",
-      "Wall time: 43 ms\n"
+      "CPU times: user 51.9 ms, sys: 46.6 ms, total: 98.6 ms\n",
+      "Wall time: 61.5 ms\n"
      ]
     },
     {
@@ -159,7 +159,7 @@
        "('batman', tensor(0), tensor([0.9405, 0.0028, 0.0566]))"
       ]
      },
-     "execution_count": 54,
+     "execution_count": 4,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -171,12 +171,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 58,
+   "execution_count": 5,
    "id": "12b3519f",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:55:02.653361Z",
-     "start_time": "2023-06-04T03:55:02.646931Z"
+     "end_time": "2023-06-04T04:18:14.466326Z",
+     "start_time": "2023-06-04T04:18:14.460213Z"
     }
    },
    "outputs": [
@@ -217,12 +217,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 59,
+   "execution_count": 6,
    "id": "0be42346",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:55:03.201333Z",
-     "start_time": "2023-06-04T03:55:03.154678Z"
+     "end_time": "2023-06-04T04:18:15.324202Z",
+     "start_time": "2023-06-04T04:18:15.281352Z"
     }
    },
    "outputs": [
@@ -272,7 +272,7 @@
        " 'Bruce Wayne')"
       ]
      },
-     "execution_count": 59,
+     "execution_count": 6,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -283,12 +283,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 60,
-   "id": "5252d05c",
+   "execution_count": 7,
+   "id": "826388a1",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:55:06.297996Z",
-     "start_time": "2023-06-04T03:55:05.937661Z"
+     "end_time": "2023-06-04T04:18:16.173080Z",
+     "start_time": "2023-06-04T04:18:15.829127Z"
     }
    },
    "outputs": [
@@ -308,12 +308,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 62,
+   "execution_count": 8,
    "id": "d845ec5f",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:55:24.063845Z",
-     "start_time": "2023-06-04T03:55:23.091052Z"
+     "end_time": "2023-06-04T04:18:18.233242Z",
+     "start_time": "2023-06-04T04:18:16.393046Z"
     }
    },
    "outputs": [
@@ -321,7 +321,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Running on local URL:  http://127.0.0.1:7867\n",
+      "Running on local URL:  http://127.0.0.1:7860\n",
       "\n",
       "To create a public link, set `share=True` in `launch()`.\n"
      ]
@@ -329,7 +329,7 @@
     {
      "data": {
       "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7867/\" 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:7860/\" 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 +342,7 @@
      "data": {
       "text/plain": []
      },
-     "execution_count": 62,
+     "execution_count": 8,
      "metadata": {},
      "output_type": "execute_result"
     },
@@ -373,6 +373,43 @@
      "metadata": {},
      "output_type": "display_data"
     },
+    {
+     "data": {
+      "text/html": [],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "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": [],
@@ -413,12 +450,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 63,
+   "execution_count": 9,
    "id": "b942385f",
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2023-06-04T03:55:42.230480Z",
-     "start_time": "2023-06-04T03:55:42.218268Z"
+     "end_time": "2023-06-04T04:18:28.883975Z",
+     "start_time": "2023-06-04T04:18:28.532380Z"
     }
    },
    "outputs": [

app.py

@@ -38,7 +38,7 @@ def classify_image(img):
     
     return prediction, alter_ego
 
-# %% app.ipynb 10
+# %% app.ipynb 9
 examples = [
     'images/batman.jpg', 'images/batman2.jpg', 'images/batman3.png', 
     'images/superman1.jpg', 'images/superman2.jpg', 'images/superman3.jpg',
@@ -48,7 +48,7 @@ examples = [
 intf = gr.Interface(
     fn=classify_image, 
     inputs=gr.Image(shape=(192,192)), 
-    outputs=gr.Label(label='Predicted output'), 
+    outputs=[gr.Label(label='Predicted output'), gr.Text(label="Alter Ego")], 
     examples=examples,
     title="Who is the 'Super Hero' Classifier",
     description="Classifier is fine-tuned on pre-trained **resnet18** model using ~200 images in total"