Upload Dockerfile

Dockerfile

@@ -13,8 +13,14 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the rest of the application into the container at /app
 COPY . /app
 
+# Create a cache directory for transformers
+RUN mkdir -p /app/cache
+
+# Set environment variable for transformers cache
+ENV TRANSFORMERS_CACHE=/app/cache
+
 # Expose port 7860
 EXPOSE 7860
 
 # Run the FastAPI application on port 7860
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]