fix error

Dockerfile

@@ -1,10 +1,13 @@
-FROM python:3.9-slim
+FROM python:3.9
 
 WORKDIR /app
 
-COPY . /app
-
-RUN pip install --upgrade pip
+COPY requirements.txt .
 RUN pip install -r requirements.txt
 
+COPY . .
+
+# Download models at build time
+RUN python download_models.py
+
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

download_models.py

@@ -0,0 +1,6 @@
+from transformers import pipeline
+
+# This will download and cache the model
+print("Downloading toxic-bert model...")
+_ = pipeline("text-classification", model="unitary/toxic-bert")
+print("Done.")

image_moderator.py

@@ -68,7 +68,7 @@ from transformers import pipeline
 from better_profanity import profanity
 
 # Load profanity model
-toxic_classifier = pipeline("text-classification", model="unitary/toxic-bert")
+toxic_classifier = pipeline("text-classification", model="unitary/toxic-bert", cache_dir="./")
 profanity.load_censor_words()
 custom_words = [
     "idiot", "moron", "dumb", "stupid", "loser", "bastard", "retard", "scumbag",