Update Dockerfile

Dockerfile

@@ -11,7 +11,8 @@ WORKDIR /code
 # RUN apt-get update && apt-get install -y --no-install-recommends some-package && rm -rf /var/lib/apt/lists/*
 
 # 4. Copy the requirements file into the container
-COPY requirements.txt .
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # 5. Install Python dependencies
 # Upgrade pip and install requirements, ensuring CPU PyTorch is used
@@ -20,8 +21,7 @@ RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
 # 6. Copy your application code and model file into the container
-COPY app.py .
-COPY model.pth .
+COPY . .
 # Add any other necessary files/folders here (e.g., utility scripts, templates)
 # COPY utils/ ./utils/