Update Dockerfile

Dockerfile

@@ -1,14 +1,19 @@
 FROM python:3.9-slim
-WORKDIR /code
 
-RUN mkdir -p /.cache && \
-    chmod 777 /.cache && \
-    mkdir -p /root/.cache/torch/sentence_transformers && \
-    chmod -R 777 /root/.cache
+WORKDIR /code
 
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
 COPY app.py .
 
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    build-essential \
+    python3-dev && \
+    pip install --no-cache-dir -r requirements.txt && \
+    apt-get remove -y build-essential python3-dev && \
+    apt-get autoremove -y && \
+    rm -rf /var/lib/apt/lists/*
+
+EXPOSE 7860
+
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]