Update Dockerfile

Dockerfile

@@ -1,12 +1,32 @@
-FROM python:3
+FROM python:3.9  # Use a specific Python version for better compatibility
+
 WORKDIR /data/app
-# Download the model file
-RUN apt-get update && apt-get install -y wget \
-    && wget -O /data/app/sneaker_category_predictor_v2.h5 "https://huggingface.co/thejagstudio/SneakerAI/resolve/main/sneaker_category_predictor_v2.h5?download=true" \
+
+# Install system dependencies required for TensorFlow
+RUN apt-get update && apt-get install -y \
+    wget \
+    python3-dev \
+    python3-pip \
+    build-essential \
+    libopenblas-dev \
+    liblapack-dev \
+    libx11-dev \
+    libglib2.0-0 \
     && apt-get clean
-RUN pip install tensorflow
+
+# Upgrade pip and install TensorFlow
+RUN pip install --upgrade pip && pip install tensorflow
+
+# Copy and install dependencies
 COPY requirements.txt ./
 RUN pip install -r requirements.txt
+
+# Download the model file
+RUN wget -O /data/app/sneaker_category_predictor_v2.h5 "https://huggingface.co/thejagstudio/SneakerAI/resolve/main/sneaker_category_predictor_v2.h5?download=true"
+
+# Copy application files
 COPY . .
+
 EXPOSE 7860
-CMD ["python","./main.py"]
+
+CMD ["python", "./main.py"]