Update Dockerfile

Dockerfile

@@ -18,6 +18,9 @@ RUN apt-get update -y && \
 # Ensure /app is writable by the non-root user
 RUN chown user:user /app
 
+# Create a directory for model weights
+RUN mkdir -p /app/models && chown user:user /app/models
+
 # Switch back to the non-root user
 USER user
 
@@ -25,8 +28,9 @@ USER user
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Copy the rest of the application code
+# Copy the rest of the application code and model weights
 COPY --chown=user . /app
+COPY --chown=user ./models/resnet50-0676ba61.pth /app/models/resnet50-0676ba61.pth
 
 # Set the command to run the application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]