Update Dockerfile

Dockerfile

@@ -5,11 +5,14 @@ WORKDIR /python-docker
 COPY requirements.txt requirements.txt
 RUN apt-get update && apt-get install git -y
 RUN pip3 install -r requirements.txt
-RUN pip3 install "git+https://github.com/openai/whisper.git" 
 RUN apt-get update && apt-get install -y ffmpeg
 
+# Set environment variable for cache directory
+ENV XDG_CACHE_HOME=/python-docker/.cache
+
+RUN python -c "from transformers import pipeline; pipeline('automatic-speech-recognition', model='openai/whisper-small')"
 COPY . .
 
-EXPOSE 8000
+EXPOSE 8001
 
-CMD ["uvicorn", "fastapi_app:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "fastapi_app:app", "--host", "0.0.0.0", "--port", "8001"]