Update Dockerfile

Dockerfile

@@ -1,33 +1,30 @@
-# # Use an official Python runtime as a parent image
+# Use an official Python runtime as a parent image
 FROM python:3.9
 
-# # Set the working directory in the container
+# Set the working directory in the container
 WORKDIR /code
 
-# # Copy the requirements file into the container
+# Copy the requirements file into the container
 COPY ./requirements.txt /code/requirements.txt
 
-# # Install any needed dependencies
+# Install any needed dependencies
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-# # Install gradio
+# Install gradio
 RUN pip install gradio
 
-# # Copy the rest of the application code into the container
+# Set the TRANSFORMERS_CACHE environment variable
+ENV TRANSFORMERS_CACHE /transformers_cache
+
+# Create the cache directory and adjust permissions
+RUN mkdir -p $TRANSFORMERS_CACHE \
+    && chmod -R 777 $TRANSFORMERS_CACHE
+
+# Copy the rest of the application code into the container
 COPY . .
 
-# # Expose the port that the FastAPI application will run on
+# Expose the port that the FastAPI application will run on
 EXPOSE 7860
 
-# # Command to run the FastAPI application with Gradio
+# Command to run the FastAPI application with Gradio
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
-
-
-# FROM tiangolo/uvicorn-gunicorn-fastapi:python3.8
-
-# COPY 
-
-# RUN pip install --no-cache-dir transformers gradio
-
-# CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "80"]
-