Added using shared learner

Deadpool-Detector.ipynb

@@ -37,68 +37,6 @@
     "Using [SH2022 Dataset](https://www.kaggle.com/datasets/muhammadhananasghar/sh2022-dataset) there is a labeled set of Deadpool images."
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 102,
-   "id": "61850308-6410-412b-a0a3-2a9c8c5ac982",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "#|default_exp app"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 103,
-   "id": "3e665c52-32ba-40a7-b100-480e60d128f6",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from fastai.vision.all import *"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "6697efd8-df96-4db1-834a-9136c6d71ebb",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# copy test data to project if not there already\n",
-    "data_path = Path(\"/home/zeus/data/SH2022/\")\n",
-    "data_path.ls()"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "04538e6b-bc2e-4ef6-b5b7-3134185ba01c",
-   "metadata": {},
-   "source": [
-    "Load the data"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "7e40d0c6-7118-4b31-8304-f0fdb1f3c490",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "train_files = get_image_files(data_path/\"train\")\n",
-    "len(train_files)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "f3e72099-446d-4e7b-8a6d-a6111f6c8556",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "test_files = get_image_files(data_path/\"test/Deadpool\")\n",
-    "len(test_files)"
-   ]
-  },
   {
    "cell_type": "markdown",
    "id": "2549da81-e133-4f5d-93e2-e754885d9d99",
@@ -114,6 +52,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from fastai.vision.all import *\n",
+    "\n",
+    "data_path = Path('/home/zeus/data/SH2022')\n",
+    "\n",
     "dls = ImageDataLoaders.from_folder(\n",
     "    path=data_path,\n",
     "    train='train',\n",
@@ -162,38 +104,170 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
-   "id": "61b3019d-cd37-49b4-bced-503499796d6c",
+   "id": "792d8a70",
    "metadata": {},
    "source": [
-    "# Create App"
+    "## Share Learner with the world!"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 104,
-   "id": "f678e4ce-6b99-45b2-ae12-3e3a6978fe0a",
+   "execution_count": 4,
+   "id": "27376bb5",
    "metadata": {},
    "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "14796a3e",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "491daa3086dd4a109f036d62bdecf5fa",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Upload 1 LFS files:   0%|          | 0/1 [00:00<?, ?it/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "46bae672442548c18d955a6d8c7ed5ad",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "model.pkl:   0%|          | 0.00/88.7M [00:00<?, ?B/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/plain": [
+       "'https://huggingface.co/jeffreymjohnson/Deadpool-Detector/tree/main/'"
+      ]
+     },
+     "execution_count": 5,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
-    "#|export\n",
-    "from fastai.vision.all import *"
+    "from huggingface_hub import push_to_hub_fastai\n",
+    "\n",
+    "learn = load_learner('/home/zeus/projects/Deadpool-Detector/deadpool-detection-model.pkl')\n",
+    "repo_id = \"jeffreymjohnson/Deadpool-Detector\"\n",
+    "push_to_hub_fastai(learner=learn, repo_id=repo_id)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "61b3019d-cd37-49b4-bced-503499796d6c",
+   "metadata": {},
+   "source": [
+    "# Create App"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 105,
-   "id": "cebedb5a-32ed-483d-90ce-9ccf101bda42",
+   "execution_count": 7,
+   "id": "f678e4ce-6b99-45b2-ae12-3e3a6978fe0a",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "a5c9312bd4be458e8694df43a510adcc",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Fetching 4 files:   0%|          | 0/4 [00:00<?, ?it/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "86445259808342c7895ef51c3c30a27b",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Downloading (…)3b40e/.gitattributes:   0%|          | 0.00/1.48k [00:00<?, ?B/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "0f5c6206271041c2b3d978afff8af373",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Downloading (…)4d02d3b40e/README.md:   0%|          | 0.00/768 [00:00<?, ?B/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "576a45913eff4216bac8db9e6a3bd8f5",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Downloading (…)3b40e/pyproject.toml:   0%|          | 0.00/162 [00:00<?, ?B/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "c847d69e1eab41b6bc51f1e65124ed5f",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Downloading model.pkl:   0%|          | 0.00/88.7M [00:00<?, ?B/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
    "source": [
     "#|export\n",
-    "learn = load_learner('/home/zeus/projects/Deadpool-Detector/deadpool-detection-model.pkl')"
+    "from huggingface_hub import from_pretrained_fastai\n",
+    "\n",
+    "learner = from_pretrained_fastai(\"jeffreymjohnson/Deadpool-Detector\")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 106,
+   "execution_count": 8,
    "id": "cc0fa8a2-51fe-4591-9056-e84e0fec9af7",
    "metadata": {},
    "outputs": [],
@@ -216,7 +290,7 @@
     "    else:\n",
     "        x = \"is not\"\n",
     "    \n",
-    "    print(\"The probability is: [\" + str(probability) + \"] that this \" + x + \" Deadpool\")\n",
+    "    #print(\"The probability is: [\" + str(probability) + \"] that this \" + x + \" Deadpool\")\n",
     "    \n",
     "    if probability < .75:\n",
     "        return PILImage.create(what_reaction)\n",
@@ -228,7 +302,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 107,
+   "execution_count": 9,
    "id": "3cec4455-8159-457b-9563-7e41aa202dc1",
    "metadata": {
     "tags": []
@@ -238,8 +312,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Running on local URL:  http://127.0.0.1:7905\n",
-      "Running on public URL: https://db2b51c5e39e6fc032.gradio.live\n",
+      "Running on local URL:  http://127.0.0.1:7860\n",
+      "Running on public URL: https://ed025bd9d3a5e5798f.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"
      ]
@@ -247,7 +321,7 @@
     {
      "data": {
       "text/html": [
-       "<div><iframe src=\"https://db2b51c5e39e6fc032.gradio.live\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
+       "<div><iframe src=\"https://ed025bd9d3a5e5798f.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>"
@@ -255,6 +329,47 @@
      },
      "metadata": {},
      "output_type": "display_data"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The probability is: [0.9999980926513672] that this is Deadpool\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"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The probability is: [0.9999998807907104] that this is not Deadpool\n"
+     ]
     }
    ],
    "source": [

app.py

@@ -1,7 +1,7 @@
 from huggingface_hub import from_pretrained_fastai
 import gradio as gr
 
-learn = from_pretrained_fastai('./deadpool-detection-model.pkl')
+learn = from_pretrained_fastai("jeffreymjohnson/Deadpool-Detector")
 
 labels = learn.dls.vocab
 
@@ -21,7 +21,6 @@ def predict(img):
     else:
         x = "is not"
     
-    print("The probability is: [" + str(probability) + "] that this " + x + " Deadpool")
     
     if probability < .75:
         return PILImage.create(what_reaction)