|
#!/bin/bash |
|
set -e |
|
|
|
|
|
mkdir -p /home/user/.cache |
|
mkdir -p /home/user/huggingface |
|
mkdir -p /app/screenshots |
|
mkdir -p /app/models |
|
|
|
|
|
chown -R user:user /home/user |
|
chown -R user:user /app/screenshots |
|
chown -R user:user /app/models |
|
|
|
|
|
if [ ! -d "/home/user/huggingface/models--indobenchmark--indobert-base-p1" ]; then |
|
echo "Pre-downloading tokenizer..." |
|
python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('indobenchmark/indobert-base-p1', use_fast=False, cache_dir='/home/user/huggingface')" |
|
fi |
|
|
|
|
|
exec python app.py |