Update Dockerfile

Dockerfile

@@ -4,6 +4,7 @@ FROM python:3.9-slim
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
+ENV TORCH_HOME=/app/.cache  # Set the torch cache directory
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -15,6 +16,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 # Set the working directory
 WORKDIR /app
 
+# Create necessary directories and set permissions
+RUN mkdir -p /app/.cache && chmod -R 777 /app
+
 # Copy requirements.txt into the container
 COPY requirements.txt .
 
@@ -24,8 +28,11 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the application code
 COPY . .
 
+# Set permissions for all files and directories in /app
+RUN chmod -R 777 /app
+
 # Expose the port for the FastAPI application
 EXPOSE 7860
 
 # Command to run the FastAPI app
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]