[Updated] docker image

Dockerfile

@@ -9,6 +9,9 @@ RUN apt-get update && apt-get install -y \
     python3-pip \
     && rm -rf /var/lib/apt/lists/*
 
+ENV TRANSFORMERS_CACHE=/app/cache
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+
 # Copy the required files
 COPY requirements.txt requirements.txt
 COPY app.py app.py
@@ -20,4 +23,4 @@ RUN pip install --no-cache-dir -r requirements.txt
 EXPOSE 7860
 
 # Run the application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,3 +1,4 @@
+import os
 import tempfile
 
 import numpy as np
@@ -5,6 +6,8 @@ import torch
 from fastapi import FastAPI, File, UploadFile
 from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
 
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+
 app = FastAPI()
 
 # Device configuration