docker compose working

Author: JurijWollert

.dockerignore

@@ -6,6 +6,8 @@
 data/
 docker/
 nbs/
+models/
+db/
 
 .git/
 .github/

.gitignore

@@ -160,3 +160,8 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+# Databases and Models
+*.db
+models/*
+!models/.gitkeep

docker-compose.yml

@@ -9,8 +9,6 @@ services:
     volumes:
       - ./db:/app/db
       - ./models:/app/models     # Mount machine learning model folder
-    environment:
-      - ENV=docker-compose
 
   frontend:
     build:
@@ -20,3 +18,5 @@ services:
       - "8501:8501"
     depends_on:
       - backend
+    environment:
+      - ENV=docker-compose

models/.gitkeep

@@ -71,12 +71,14 @@ def main() -> None:
                     "tags": ",".join(tags) if tags else "",
                 },
                 files={"image_bytes": ("image.jpg", buffer, "image/jpeg")},
+                timeout=10,
             ).json()["record-id"]
 
             result = httpx.post(
                 f"{BACKEND_URL}/predict",
                 data={"record_id": record_id},
                 files={"file": ("image.jpg", buffer, "image/jpeg")},
+                timeout=10,
             ).json()["prediction"]
 
         st.header(f"Assigned Institution: {result}")