Update Dockerfile

Dockerfile

@@ -1,25 +1,41 @@
 FROM python:3.10-slim
 
-# Install system dependencies
+# Install system dependencies and build tools
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     git \
     curl \
- && apt-get clean \
- && rm -rf /var/lib/apt/lists/*
+    build-essential \
+    cmake \
+    libgomp1 \
+ && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-# Set work directory
+# Set working directory
 WORKDIR /app
 
-# Copy files
+# Copy application files
 COPY . /app
 
 # Install Python dependencies
 RUN pip install --upgrade pip \
- && pip install -r requirements.txt
+ && pip install -r requirements.txt \
+ && pip install huggingface_hub
 
-# Expose the port for Flask
+# Download GGUF model at build time into models/
+RUN mkdir -p models \
+ && python - <<EOF
+from huggingface_hub import hf_hub_download
+hf_hub_download(
+    repo_id="TheBloke/Mistral-7B-Instruct-GPTQ",
+    filename="mistral-7b-instruct-q4_k_m.gguf",
+    cache_dir="models",
+    library_name="llama-cpp-python"
+)
+EOF
+
+# Expose port 7860 for Flask app
 EXPOSE 7860
 
-# Run the app
+# Start the Flask application
 CMD ["python", "app.py"]
+```