Upload 3 files

Dockerfile

@@ -0,0 +1,29 @@
+FROM python:3.11-bookworm
+
+COPY requirements.txt requirements.txt
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+      bzip2 \
+      g++ \
+      git \
+      graphviz \
+      libgl1-mesa-glx \
+      libhdf5-dev \
+      openmpi-bin \
+      wget \
+      python3-tk && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN pip install --upgrade pip
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+RUN useradd -m -u 1000 myuser
+
+USER myuser
+
+COPY --chown=myuser app app
+
+EXPOSE 8001
+
+CMD ["python", "app/main.py"]

face_recognition_model.py

@@ -0,0 +1,31 @@
+import math
+import torch
+import torchvision
+import torch.nn as nn
+import torch.nn.functional as F
+from torchvision import transforms
+# Add more imports if required
+
+# Sample Transformation function
+# YOUR CODE HERE for changing the Transformation values.
+trnscm = transforms.Compose([transforms.Resize((100,100)), transforms.ToTensor()])
+
+##Example Network
+class Siamese(torch.nn.Module):
+    def __init__(self):
+        super(Siamese, self).__init__()
+        #YOUR CODE HERE
+        
+    def forward(self, x):
+        pass   # remove 'pass' once you have written your code
+        #YOUR CODE HERE
+        
+##########################################################################################################
+## Sample classification network (Specify if you are using a pytorch classifier during the training)    ##
+## classifier = nn.Sequential(nn.Linear(64, 64), nn.BatchNorm1d(64), nn.ReLU(), nn.Linear...)           ##
+##########################################################################################################
+
+# YOUR CODE HERE for pytorch classifier
+
+# Definition of classes as dictionary
+classes = ['person1','person2','person3','person4','person5','person6','person7']

requirements.txt

@@ -0,0 +1,19 @@
+uvicorn==0.17.6
+fastapi==0.99.1
+pydantic==1.10.10
+requests==2.23.0
+jinja2==3.1.2
+python-multipart==0.0.6
+
+scikit-learn==1.2.2
+joblib==1.3.2
+Pillow==9.4.0
+torch==2.1.0
+torchvision==0.16.0
+matplotlib==3.7.1
+
+#opencv-python==4.5.5.64
+numpy==1.26.4
+pandas==2.2.2
+# Prefer headless on servers (no GUI deps); 4.9 works well with numpy 1.26.x
+opencv-python-headless==4.9.0.80