From 1d68bc889f1075a167be1ac2a98378c5c3bb74c8 Mon Sep 17 00:00:00 2001
From: Mostafa Moradian <mostafa@gatewayd.io>
Date: Fri, 24 May 2024 22:47:00 +0200
Subject: [PATCH] Add temporary solution for running Serving API

---
 docker-compose.yaml | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index bb6b184..207e767 100644
--- 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"