Update Dockerfile

Dockerfile

@@ -18,7 +18,7 @@ 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
+# Create a directory for model weights (optional, if you plan to add models later)
 RUN mkdir -p /app/models && chown user:user /app/models
 
 # Switch back to the non-root user
@@ -28,9 +28,8 @@ 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 and model weights
+# Copy the rest of the application code
 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"]