Update Dockerfile

Dockerfile

@@ -4,19 +4,13 @@ FROM python:3.10-slim
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
-    HF_HOME=/home/user/.cache/huggingface \
-    TRANSFORMERS_CACHE=/home/user/.cache/huggingface/transformers \
+    HF_HOME=/app/.cache \
     TRANSFORMERS_VERBOSITY=error
 
-# Create a non-root user
-RUN useradd -m -u 1000 user
-
-# Set working directory
+# Set working directory and create necessary folders
 WORKDIR /app
-
-# Create necessary folders with proper permissions
-RUN mkdir -p /home/user/.cache/huggingface /app/models/suno-bark /app/models/sentiment && \
-    chmod -R 777 /home/user/.cache /app/models
+RUN mkdir -p /app/.cache /app/models/suno-bark /app/models/sentiment && \
+    chmod -R 777 /app/.cache /app/models
 
 # Install OS dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -31,16 +25,23 @@ RUN pip install --no-cache-dir --root-user-action=ignore -U pip && \
         "protobuf" && \
     pip install --no-cache-dir --root-user-action=ignore -r requirements.txt
 
-# Ensure soundfile is installed
+# Install soundfile
 RUN pip install soundfile
 
-# Set user to non-root for Hugging Face compatibility
-USER user
+# Download Bark TTS model using snapshot_download
+RUN python3 -c "\
+from huggingface_hub import snapshot_download; \
+snapshot_download(repo_id='suno/bark-small', local_dir='/app/models/suno-bark', local_dir_use_symlinks=False)"
+
+# Download sentiment analysis model using snapshot_download
+RUN python3 -c "\
+from huggingface_hub import snapshot_download; \
+snapshot_download(repo_id='cardiffnlp/twitter-xlm-roberta-base-sentiment', local_dir='/app/models/sentiment', local_dir_use_symlinks=False)"
 
 # Copy application source code
-COPY --chown=user:user app.py .
+COPY app.py .
 
-# Expose port
+# Expose port for the application
 EXPOSE 7860
 
 # Launch app using Uvicorn