Navya-Sree's picture
Update setup.sh
e92b857 verified
raw
history blame contribute delete
331 Bytes
#!/bin/bash
# Install Python dependencies
pip install -r requirements.txt
# Download the model during build (prevents timeout on first request)
python -c "from transformers import AutoTokenizer, AutoModelForSeq2SeqLM; AutoTokenizer.from_pretrained('tscholak/cxmefzzi'); AutoModelForSeq2SeqLM.from_pretrained('tscholak/cxmefzzi')"