Update Dockerfile

Dockerfile

@@ -1,24 +1,23 @@
-FROM python:3.13.5-slim
-
-WORKDIR /app
-
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    git \
-    && rm -rf /var/lib/apt/lists/*
-
-COPY requirements.txt ./
-COPY src/ ./src/
-COPY resnet18_cifar10_finetuned.pth ./
-
-RUN pip3 install -r requirements.txt
-
-# Change EXPOSE to 7860
-EXPOSE 7860
-
-# Update HEALTHCHECK to use 7860
-HEALTHCHECK CMD curl --fail http://localhost:7860/_stcore/health
-
-# Add the XSRF and CORS disable flags to ENTRYPOINT
-ENTRYPOINT ["streamlit", "run", "src/image_classifier_app.py", "--server.port=7860", "--server.address=0.0.0.0", "--server.enableCORS=false", "--server.enableXsrfProtection=false"]
+FROM python:3.13.5-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt ./
+COPY ..
+
+RUN pip3 install -r requirements.txt
+
+# Change EXPOSE to 7860
+EXPOSE 7860
+
+# Update HEALTHCHECK to use 7860
+HEALTHCHECK CMD curl --fail http://localhost:7860/_stcore/health
+
+# Add the XSRF and CORS disable flags to ENTRYPOINT
+ENTRYPOINT ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0", "--server.enableCORS=false", "--server.enableXsrfProtection=false"]