Update Dockerfile

Dockerfile

@@ -4,6 +4,7 @@ WORKDIR /app
 
 # System deps
 RUN apt-get update && apt-get install -y \
+    ffmpeg \
     git \
     curl \
     && rm -rf /var/lib/apt/lists/*
@@ -12,10 +13,10 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Pre-download model (IMPORTANT)
+# Pre-download model (prevents runtime failures)
 RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
-AutoTokenizer.from_pretrained('distilgpt2'); \
-AutoModelForCausalLM.from_pretrained('distilgpt2')"
+AutoTokenizer.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0'); \
+AutoModelForCausalLM.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0')"
 
 # Copy app
 COPY app.py .