Spaces:

saadkhi
/

SQL_chatbot_API

Running

Update Dockerfile

b8d478f verified 7 months ago

424 Bytes

FROM python:3.10

ENV HF_HOME=/app/cache

WORKDIR /app

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# 👇 Pre-download model
RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
AutoTokenizer.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0'); \
AutoModelForCausalLM.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0')"

COPY app.py .

CMD ["python", "app.py"]

	FROM python:3.10

	ENV HF_HOME=/app/cache

	WORKDIR /app

	COPY requirements.txt .
	RUN pip install --no-cache-dir -r requirements.txt

	# 👇 Pre-download model
	RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
	AutoTokenizer.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0'); \
	AutoModelForCausalLM.from_pretrained('TinyLlama/TinyLlama-1.1B-Chat-v1.0')"

	COPY app.py .

	CMD ["python", "app.py"]