fixed live demo nb

Author: jaybdub

notebooks/resnet18/live_demo.ipynb

@@ -38,7 +38,7 @@
    "source": [
     "from jetcam.utils import bgr8_to_jpeg\n",
     "import traitlets\n",
-    "import ipywidgetsb\n",
+    "import ipywidgets\n",
     "\n",
     "image_w = ipywidgets.Image()\n",
     "\n",
@@ -60,6 +60,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import torch\n",
     "from torch2trt import TRTModule\n",
     "\n",
     "model_trt = TRTModule()\n",
@@ -81,6 +82,7 @@
    "source": [
     "import cv2\n",
     "import numpy as np\n",
+    "import torchvision\n",
     "\n",
     "device = torch.device('cuda')\n",
     "mean = 255.0 * np.array([0.485, 0.456, 0.406])\n",