Update Dockerfile

Dockerfile

@@ -1,21 +1,24 @@
 FROM python:3.9-slim
 
-WORKDIR /app
-
+# Install dependencies
 RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
+    gcc \
+    g++ \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt ./
-COPY src/ ./src/
+# Set working directory
+WORKDIR /app
 
-RUN pip3 install -r requirements.txt
+# Copy requirements and install
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-EXPOSE 8501
+# Copy application files
+COPY app.py .
+COPY model.pkl .
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Expose Streamlit port
+EXPOSE 8501
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Run Streamlit
+CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]