update app model

app.ipynb

@@ -75,7 +75,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 68,
+   "execution_count": 77,
    "metadata": {},
    "outputs": [
     {
@@ -125,7 +125,7 @@
        "             2.7985e-07, 8.2198e-09]))"
       ]
      },
-     "execution_count": 68,
+     "execution_count": 77,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -136,12 +136,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 69,
+   "execution_count": 79,
    "metadata": {},
    "outputs": [],
    "source": [
     "#|export\n",
-    "categories = ('Jacaranda', 'Bougainvillea', 'Clavalina tree', 'Birds of paradise plant oaxaca', 'Flame vine', 'Plumeria rubra', 'Ceiba speciosa', 'Poinsettia tree', 'Calistemo', 'Flamboyan', 'Mexican creeper','primavera rosa')\n",
+    "categories = learn.dls.vocab\n",
     "\n",
     "def classify_image(img):\n",
     "    pred,idx,probs = learn.predict(img)\n",
@@ -193,7 +193,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 71,
+   "execution_count": 82,
    "metadata": {},
    "outputs": [
     {
@@ -229,13 +229,12 @@
     "\n",
     "# Print the sorted list of photos\n",
     "print(photos)\n",
-    "\n",
-    "example_photo=['Poinsettia tree.jpg', 'Mexican creeper.jpg', 'Plumeria rubra.jpg']\n"
+    "\n"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 75,
+   "execution_count": 81,
    "metadata": {},
    "outputs": [
     {
@@ -258,55 +257,18 @@
      "text": [
       "IMPORTANT: You are using gradio version 3.9, however version 3.14.0 is available, please upgrade.\n",
       "--------\n",
-      "Running on local URL:  http://127.0.0.1:7875\n",
+      "Running on local URL:  http://127.0.0.1:7876\n",
       "\n",
       "To create a public link, set `share=True` in `launch()`.\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"
-    },
     {
      "data": {
       "text/plain": [
-       "(<gradio.routes.App at 0x2a41a6110>, 'http://127.0.0.1:7875/', None)"
+       "(<gradio.routes.App at 0x2a41d6d40>, 'http://127.0.0.1:7876/', None)"
       ]
      },
-     "execution_count": 75,
+     "execution_count": 81,
      "metadata": {},
      "output_type": "execute_result"
     },
@@ -411,117 +373,6 @@
      "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"
-      ],
-      "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": [],
@@ -539,7 +390,7 @@
     "label = gr.outputs.Label()\n",
     "examples = photos\n",
     "\n",
-    "intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=example_photo)\n",
+    "intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)\n",
     "intf.launch(inline=False)"
    ]
   }

app.py

@@ -9,7 +9,7 @@ learn = load_learner('mexicanPlants (1).pkl')
 
 
 #|export
-categories = ('Jacaranda', 'Bougainvillea', 'Clavalina tree', 'Birds of paradise plant oaxaca', 'Flame vine', 'Plumeria rubra', 'Ceiba speciosa', 'Poinsettia tree', 'Calistemo', 'Flamboyan', 'Mexican creeper','primavera rosa')
+categories = learn.dls.vocab
 
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
@@ -31,7 +31,7 @@ photos = []
 for file in all_files:
     # Check if the file is a photo
     if file.endswith(('.jpg', '.jpeg', '.png', '.bmp', '.gif')):
-        # If it is, add it to the list of photos
+        # If it is, add it to the list of photosgi
         photos.append(file)
 
 image = gr.inputs.Image(shape=(192,192))