From 18532144098ac159545d6bc6b48547ee08b1e716 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 15 Apr 2023 13:18:24 -0500
Subject: [PATCH] fix(build): install CPU deps during CI

---
 .gitlab-ci.yml       | 2 ++
 api/Makefile         | 1 +
 api/onnx_web/main.py | 5 +++++
 3 files changed, 8 insertions(+)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 03e1efa1..46d51eaa 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -59,6 +59,8 @@ stages:
     key:
       files:
         - api/requirements/base.txt
+        - api/requirements/cpu.txt
+        - api/requirements/dev.txt
     paths:
       - .cache/pip
     policy: pull-push
diff --git a/api/Makefile b/api/Makefile
index 1ee72d01..283c1ef1 100644
--- a/api/Makefile
+++ b/api/Makefile
@@ -10,6 +10,7 @@ check-venv:
 	if [ -z $${VIRTUAL_ENV+x} ]; then echo "Are you sure you want to install dependencies outside of a virtual environment?"; sleep 30; fi
 
 pip: check-venv
+	pip install -r requirements/cpu.txt
 	pip install -r requirements/base.txt
 
 pip-dev: check-venv
diff --git a/api/onnx_web/main.py b/api/onnx_web/main.py
index 65c15547..05143853 100644
--- a/api/onnx_web/main.py
+++ b/api/onnx_web/main.py
@@ -1,5 +1,6 @@
 import atexit
 import gc
+import mimetypes
 from functools import partial
 from logging import getLogger
 
@@ -32,6 +33,10 @@ def main():
     setproctitle("onnx-web server")
     set_start_method("spawn", force=True)
 
+    # set up missing mimetypes
+    mimetypes.add_type("application/javascript", ".js")
+    mimetypes.add_type("text/css", ".css")
+
     context = ServerContext.from_environ()
     apply_patches(context)
     check_paths(context)