Update Dockerfile

Dockerfile

@@ -1,25 +1,17 @@
-# ---- Base Image ----
+# Use lightweight Python image with CUDA support (if needed)
 FROM python:3.10-slim
 
-# ---- System Dependencies ----
-RUN apt-get update && apt-get install -y \
-    libgl1 libglib2.0-0 \
-    && rm -rf /var/lib/apt/lists/*
+# Install system dependencies
+RUN apt-get update && apt-get install -y libgl1 libglib2.0-0 && rm -rf /var/lib/apt/lists/*
 
-# ---- Working Directory ----
+# Copy app
 WORKDIR /app
-
-# ---- Copy Files ----
 COPY requirements.txt .
-COPY app.py .
-COPY best.pt .
-
-# ---- Install Python Dependencies ----
 RUN pip install --no-cache-dir -r requirements.txt
 
-# ---- Environment Variables ----
-ENV PORT=7860
+COPY . .
+
+# Expose Flask port
 EXPOSE 7860
 
-# ---- Start Flask App ----
 CMD ["python", "app.py"]