Update Dockerfile

Dockerfile

@@ -7,8 +7,8 @@ WORKDIR /app
 # Copy the current directory contents into the container at /app
 COPY . /app
 
-# Create a directory for the cache and set permissions
-RUN mkdir -p /app/.cache && chmod 777 /app/.cache
+# Create directories for the cache and flagged data, and set permissions
+RUN mkdir -p /app/.cache /app/flagged && chmod 777 /app/.cache /app/flagged
 
 # Install any needed packages specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
@@ -16,8 +16,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Make port 7860 available to the world outside this container
 EXPOSE 7860
 
-# Set environment variable for the cache directory
-ENV TRANSFORMERS_CACHE=/app/.cache
+# Set environment variable for the Hugging Face home directory
+ENV HF_HOME=/app/.cache
 
 # Run app.py when the container launches
 CMD ["python", "app.py"]