Update Dockerfile

Dockerfile

@@ -1,6 +1,4 @@
-# Use a base image provided by Hugging Face Spaces for Gradio SDK
-# Check HF docs for the latest recommended base image if needed
-# FROM huggingface/python-dependencies:latest
+# Use a base image (Python 3.10 slim)
 FROM python:3.10-slim
 
 # Set working directory
@@ -9,10 +7,28 @@ WORKDIR /code
 # Copy requirements first to leverage Docker cache
 COPY ./requirements.txt /code/requirements.txt
 
-# Install Python dependencies
-# Using --no-cache-dir can sometimes help in resource-constrained environments
-RUN pip install --no-cache-dir --upgrade pip
-RUN pip install --no-cache-dir -r /code/requirements.txt
+# Install system dependencies (curl for downloading), Python dependencies,
+# create /data dir, check for model and download if missing, then clean up.
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends curl && \
+    pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r /code/requirements.txt && \
+    mkdir -p /data && \
+    echo "Checking for model file in /data..." && \
+    if [ ! -f /data/zephyr-7b-beta.Q4_K_M.gguf ]; then \
+      echo "Model file not found. Downloading..." && \
+      curl -L -o /data/zephyr-7b-beta.Q4_K_M.gguf https://huggingface.co/TheBloke/zephyr-7B-beta-GGUF/resolve/main/zephyr-7b-beta.Q4_K_M.gguf && \
+      echo "Download complete."; \
+    else \
+      echo "Model file already exists."; \
+    fi && \
+    # Clean up apt cache
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
 
-# Copy the rest of the application code
+# Copy the rest of the application code from the build context to the container
 COPY . /code/
+
+# Set the command to run the Gradio app (app.py)
+# HF Spaces usually figures this out, but explicitly setting it is fine.
+# CMD ["python", "app.py"]