Add temporary solution for running Serving API

gatewayd-io · May 24, 2024 · 1d68bc8 · 1d68bc8
1 parent b29d7ca
commit 1d68bc8
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -7,10 +7,17 @@ services:
     ports:
       - "8000:8000"
   serving-api:
-    build:
-      context: .
-      dockerfile: Dockerfile.serving-api
-    image: serving-api:latest
+    # build:
+    #   context: .
+    #   dockerfile: Dockerfile.serving-api
+    # image: serving-api:latest
+    # This is a temporary solution to run the serving-api,
+    # until the serving-api is ready to be run in a container.
+    image: tensorflow/serving:latest
+    environment:
+      - MODEL_NAME=sqli_model
+    volumes:
+      - ./sqli_model:/models/sqli_model
     ports:
       - "8500:8500"
       - "8501:8501"