Update Dockerfile

Dockerfile

@@ -1,34 +1,20 @@
-# Use a minimal Python base image
 FROM python:3.10-slim
 
-# Set working directory
 WORKDIR /app
 
-# Cache env-vars
 ENV HF_HOME=/app/.cache/huggingface
 ENV XDG_CACHE_HOME=/app/.cache
 
-# Install curl (and ca-certificates) for Ollama
-RUN apt-get update && \
-    apt-get install -y curl ca-certificates && \
-    rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 
-# Install Ollama v0.3.1 (CPU-safe)
-RUN curl -fsSL https://ollama.com/download/ollama-linux-amd64-v0.3.1.tgz | \
-    tar -xz -C /usr/local --strip-components=1
-
-# Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the code
 COPY . .
 
-# Ensure start script is executable
-RUN chmod +x /app/start.sh
+# (Optional) pre-download model to image to avoid runtime download
+# RUN python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; m='google/gemma-2-2b-it'; AutoTokenizer.from_pretrained(m); AutoModelForCausalLM.from_pretrained(m)"
 
-# Expose Gradio’s default port
 EXPOSE 7860
 
-# Launch everything
-ENTRYPOINT ["bash", "/app/start.sh"]
+CMD ["python", "main.py"]