Update Dockerfile

Dockerfile

@@ -1,22 +1,21 @@
 FROM tensorflow/tensorflow:2.15.0
 
+# Don't use slim version! Full version has tf.keras bundled.
+
 WORKDIR /app
 
-# Environment flags
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 ENV TF_USE_LEGACY_KERAS=1
 
-# Copy files
 COPY . .
-
-# Ensure model is readable
 RUN chmod 644 /app/2.h5
 
-# Install dependencies
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
+# Install dependencies (no tensorflow here!)
+RUN pip install --no-cache-dir --upgrade pip \
+ && pip install --no-cache-dir -r requirements.txt
 
 EXPOSE 7860
 
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+