app complete

.gitignore

@@ -0,0 +1,2 @@
+.ipynb_checkpoints
+*.Identifier

Deadpool-Detector.ipynb

@@ -39,7 +39,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 102,
    "id": "61850308-6410-412b-a0a3-2a9c8c5ac982",
    "metadata": {},
    "outputs": [],
@@ -49,7 +49,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 103,
    "id": "3e665c52-32ba-40a7-b100-480e60d128f6",
    "metadata": {},
    "outputs": [],
@@ -171,7 +171,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 104,
    "id": "f678e4ce-6b99-45b2-ae12-3e3a6978fe0a",
    "metadata": {},
    "outputs": [],
@@ -182,7 +182,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 105,
    "id": "cebedb5a-32ed-483d-90ce-9ccf101bda42",
    "metadata": {},
    "outputs": [],
@@ -193,7 +193,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 106,
    "id": "cc0fa8a2-51fe-4591-9056-e84e0fec9af7",
    "metadata": {},
    "outputs": [],
@@ -201,23 +201,45 @@
     "#|export\n",
     "labels = learn.dls.vocab\n",
     "def predict(img):\n",
+    "    yes_reaction = \"reactions/yes.jfif\"\n",
+    "    no_reaction = \"reactions/no.jfif\"\n",
+    "    what_reaction = \"reactions/what.jpg\"\n",
+    "    \n",
     "    img = PILImage.create(img)\n",
     "    pred,pred_idx,probs = learn.predict(img)\n",
-    "    return {labels[i]: float(probs[i]) for i in range(len(labels))}"
+    "    index = pred_idx.item()\n",
+    "    probability = probs[index].item()\n",
+    "    isDeadpool = pred == \"Deadpool\"\n",
+    "    \n",
+    "    if isDeadpool:\n",
+    "        x = \"is\"\n",
+    "    else:\n",
+    "        x = \"is not\"\n",
+    "    \n",
+    "    print(\"The probability is: [\" + str(probability) + \"] that this \" + x + \" Deadpool\")\n",
+    "    \n",
+    "    if probability < .75:\n",
+    "        return PILImage.create(what_reaction)\n",
+    "    elif isDeadpool:\n",
+    "        return PILImage.create(yes_reaction)\n",
+    "    else:\n",
+    "        return PILImage.create(no_reaction)"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 107,
    "id": "3cec4455-8159-457b-9563-7e41aa202dc1",
-   "metadata": {},
+   "metadata": {
+    "tags": []
+   },
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Running on local URL:  http://127.0.0.1:7863\n",
-      "Running on public URL: https://40a72959264a859fdc.gradio.live\n",
+      "Running on local URL:  http://127.0.0.1:7905\n",
+      "Running on public URL: https://db2b51c5e39e6fc032.gradio.live\n",
       "\n",
       "This share link expires in 72 hours. For free permanent hosting and GPU upgrades (NEW!), check out Spaces: https://huggingface.co/spaces\n"
      ]
@@ -225,108 +247,7 @@
     {
      "data": {
       "text/html": [
-       "<div><iframe src=\"https://40a72959264a859fdc.gradio.live\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "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": [],
-      "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": [],
-      "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"
+       "<div><iframe src=\"https://db2b51c5e39e6fc032.gradio.live\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
       ],
       "text/plain": [
        "<IPython.core.display.HTML object>"
@@ -334,16 +255,6 @@
      },
      "metadata": {},
      "output_type": "display_data"
-    },
-    {
-     "data": {
-      "text/html": [],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
     }
    ],
    "source": [
@@ -354,7 +265,7 @@
     "    title=\"Deadpool™️ Detector\",\n",
     "    fn=predict,\n",
     "    inputs=gr.Image(shape=(200,200)),\n",
-    "    outputs=gr.Label(num_top_classes=-3),\n",
+    "    outputs=gr.Image(shape=(200,200)), \n",
     "    examples=[\"examples/deadpool_example1.jpg\", \"examples/deadpool_example2.jpg\", \"examples/deadpool_example3.jpg\", \"examples/spiderman.jpg\",]\n",
     ").launch(share=true)"
    ]
@@ -362,7 +273,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "018d3761-beac-4999-b915-32e61297db7c",
+   "id": "0c703c3c-f9a9-4c44-a1d5-44dd1d6189ae",
    "metadata": {},
    "outputs": [],
    "source": []

app.py

@@ -1,12 +1,39 @@
+from fastai.vision.all import *
 import gradio as gr
 
-def classify(input_img):
-    out_image = input_img
-    return out_image
+learn = load_learner('./deadpool-detection-model.pkl')
 
-demo = gr.Interface(
-    fn=classify, 
-    inputs=gr.Image(shape=(200,200)), 
-    outputs="image")
+labels = learn.dls.vocab
 
-demo.launch()
+def predict(img):
+    yes_reaction = "reactions/yes.jfif"
+    no_reaction = "reactions/no.jfif"
+    what_reaction = "reactions/what.jpg"
+    
+    img = PILImage.create(img)
+    pred,pred_idx,probs = learn.predict(img)
+    index = pred_idx.item()
+    probability = probs[index].item()
+    isDeadpool = pred == "Deadpool"
+    
+    if isDeadpool:
+        x = "is"
+    else:
+        x = "is not"
+    
+    print("The probability is: [" + str(probability) + "] that this " + x + " Deadpool")
+    
+    if probability < .75:
+        return PILImage.create(what_reaction)
+    elif isDeadpool:
+        return PILImage.create(yes_reaction)
+    else:
+        return PILImage.create(no_reaction)
+    
+demo = gr.Interface(
+    title="Deadpool™️ Detector",
+    fn=predict,
+    inputs=gr.Image(shape=(200,200)),
+    outputs=gr.Image(shape=(200,200)), 
+    examples=["examples/deadpool_example1.jpg", "examples/deadpool_example2.jpg", "examples/deadpool_example3.jpg", "examples/spiderman.jpg",]
+).launch(share=true)