Update Dockerfile

Dockerfile

@@ -4,6 +4,7 @@ WORKDIR /data/app
 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" \
     && apt-get clean
+RUN pip install tensorflow
 COPY requirements.txt ./
 RUN pip install -r requirements.txt
 COPY . .