changes

Dockerfile

@@ -1,13 +1,18 @@
-FROM python:3.9
+# Use a slim Python image
+FROM python:3.9-slim
 
+# Set working directory
 WORKDIR /app
 
+# Install dependencies
 COPY requirements.txt .
-RUN pip install -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
+# Copy app code and models
 COPY . .
 
-# Download models at build time
-RUN python download_models.py
+# Expose the default port (Hugging Face uses 7860, but can be any port)
+EXPOSE 7860
 
+# Start uvicorn
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

download_models.py

@@ -1,6 +0,0 @@
-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

@@ -66,9 +66,9 @@ class ImageContentModerator:
         return output_path
 from transformers import pipeline
 from better_profanity import profanity
-os.makedirs("./cache_dir", exist_ok=True)
+
 # Load profanity model
-toxic_classifier = pipeline("text-classification", model="unitary/toxic-bert", cache_dir="./cache_dir")
+pipeline("text-classification", model="unitary/toxic-bert").save_pretrained("./models/unitary-toxic-bert")
 profanity.load_censor_words()
 custom_words = [
     "idiot", "moron", "dumb", "stupid", "loser", "bastard", "retard", "scumbag",