ddd

Dockerfile

@@ -1,14 +1,25 @@
-# Use the official Python 3.10.9 image
-FROM python:3.12
+# Use a lightweight and stable Python base image
+FROM python:3.10-slim
 
-# Copy the current directory contents into the container at .
-COPY . .
+# Set environment variables to prevent cache permission issues
+ENV HF_HOME=/tmp/huggingface
+ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
+ENV PIP_NO_CACHE_DIR=1
+ENV PYTHONUNBUFFERED=1
 
-# Set the working directory to /
-WORKDIR /
+# Set working directory
+WORKDIR /app
 
-# Install requirements.txt
-RUN pip install --no-cache-dir --upgrade -r /requirements.txt
+# Copy files into the container
+COPY . /app
+
+# Install dependencies
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
+
+# Expose the default port used by Hugging Face Spaces
+EXPOSE 7860
+
+# Run the FastAPI app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
-# Start the FastAPI app on port 7860, the default port expected by Spaces
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -6,11 +6,6 @@ from happytransformer import HappyTextToText, TTSettings
 
 import os
 
-# Redirect cache to a local writable path
-os.environ["HF_HOME"] = "/tmp/huggingface"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
-os.makedirs("/tmp/huggingface", exist_ok=True)
-
 app = FastAPI()
 
 # Load models at startup