fix: Add apt packages for Pillow (#81)

VertaAI · Sep 15, 2023 · bdd13fc · bdd13fc
1 parent 42886fd
commit bdd13fc
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/deployment/tfhub/object-detection-base64.ipynb b/deployment/tfhub/object-detection-base64.ipynb
@@ -292,9 +292,9 @@
       "outputs": [],
       "source": [
         "model = registered_model.create_standard_model(\n",
-        "    name = VERSION,\n",
-        "    model_cls = DetectObject,\n",
-        "    model_api = ModelAPI(\n",
+        "    name=VERSION,\n",
+        "    model_cls=DetectObject,\n",
+        "    model_api=ModelAPI(\n",
         "        [{'file_name': '', 'img': ''}],\n",
         "        [{\n",
         "            'file_name': '',\n",
@@ -304,11 +304,14 @@
         "                'ymin': 0,\n",
         "                'xmin': 0,\n",
         "                'ymax': 0,\n",
-        "                'xmax': 0\n",
+        "                'xmax': 0,\n",
         "            }\n",
         "        }]\n",
         "    ),\n",
-        "    environment = Python(requirements = ['tensorflow', 'tensorflow_hub', 'dill', 'Pillow', 'wget'])\n",
+        "    environment=Python(\n",
+        "        requirements=['tensorflow', 'tensorflow_hub', 'dill', 'Pillow', 'wget'],\n",
+        "        apt_packages=['libjpeg-dev', 'zlib1g-dev'],  # for Pillow\n",
+        "    )\n",
         ")"
       ]
     },

diff --git a/deployment/tfhub/object-detection-url.ipynb b/deployment/tfhub/object-detection-url.ipynb
@@ -319,9 +319,9 @@
       "outputs": [],
       "source": [
         "model = registered_model.create_standard_model(\n",
-        "    name = VERSION,\n",
-        "    model_cls = DetectObject,\n",
-        "    model_api = ModelAPI(\n",
+        "    name=VERSION,\n",
+        "    model_cls=DetectObject,\n",
+        "    model_api=ModelAPI(\n",
         "        [{'file_name': '', 'url': ''}],\n",
         "        [{\n",
         "            'file_name': '',\n",
@@ -331,11 +331,14 @@
         "                'ymin': 0,\n",
         "                'xmin': 0,\n",
         "                'ymax': 0,\n",
-        "                'xmax': 0\n",
+        "                'xmax': 0,\n",
         "            }\n",
         "        }]\n",
         "    ),\n",
-        "    environment = Python(requirements = ['tensorflow', 'tensorflow_hub', 'dill', 'Pillow', 'wget'])\n",
+        "    environment=Python(\n",
+        "        requirements=['tensorflow', 'tensorflow_hub', 'dill', 'Pillow', 'wget'],\n",
+        "        apt_packages=['libjpeg-dev', 'zlib1g-dev'],  # for Pillow\n",
+        "    )\n",
         ")"
       ]
     },