Add model pre-download steps to Dockerfile to prevent timeout

Dockerfile

@@ -8,10 +8,20 @@ WORKDIR /app
 COPY requirements.txt .
 
 # Install any needed packages specified in requirements.txt
-# Also, download textblob corpora which is required by sentiment_analyzer.py
+# Also, download textblob corpora
 RUN pip install --no-cache-dir -r requirements.txt \
     && python -m textblob.download_corpora
 
+# --- ADD THESE LINES TO PRE-DOWNLOAD HUGGING FACE MODELS ---
+# You need to know the exact model IDs your sentiment and sarcasm scripts use.
+# I'll use common examples. Adjust if your models are different.
+# Example for Sentiment Model (adjust model name if yours is different):
+RUN python -c "from transformers import pipeline; pipeline('sentiment-analysis', model='cardiffnlp/twitter-roberta-base-sentiment-latest')"
+
+# Example for Sarcasm Model (adjust model name if yours is different):
+RUN python -c "from transformers import AutoTokenizer, AutoModelForSequenceClassification; tokenizer = AutoTokenizer.from_pretrained('mrm8488/distilroberta-base-finetuned-sarcasm'); model = AutoModelForSequenceClassification.from_pretrained('mrm8488/distilroberta-base-finetuned-sarcasm')"
+# --- END OF ADDED LINES ---
+
 # Copy the entire project directory into the container at /app
 COPY . .
 
@@ -20,4 +30,4 @@ EXPOSE 8501
 
 # Define the command to run the Streamlit application
 # Streamlit runs on 0.0.0.0 by default in Docker
-CMD ["streamlit", "run", "frontend/app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+CMD ["streamlit", "run", "frontend/app.py", "--server.port=8501", "--server.address=0.0.0.0"]