Import timm

app.ipynb

@@ -17,7 +17,8 @@
    "source": [
     "#|export\n",
     "from fastai.vision.all import *\n",
-    "import gradio as gr"
+    "import gradio as gr\n",
+    "import timm"
    ]
   },
   {
@@ -66,43 +67,6 @@
      "execution_count": 5,
      "metadata": {},
      "output_type": "execute_result"
-    },
-    {
-     "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"
     }
    ],
    "source": [

app.py

@@ -6,6 +6,7 @@ __all__ = ['learn', 'categories', 'image', 'label', 'path', 'examples', 'intf',
 # %% ../app.ipynb 1
 from fastai.vision.all import *
 import gradio as gr
+import timm
 
 # %% ../app.ipynb 2
 learn = load_learner("model.pkl")
@@ -21,8 +22,8 @@ def classify_img(img):
 # %% ../app.ipynb 4
 from pathlib import Path
 
-image = gr.inputs.Image(shape=(192, 192))
-label = gr.outputs.Label()
+image = gr.Image(shape=(192, 192))
+label = gr.Label()
 
 path = Path("examples")
 examples = [file for file in path.iterdir()]