automated update: 2026-06-10 21:13:39
This commit is contained in:
+9
-27
@@ -1,48 +1,30 @@
|
||||
#!/bin/bash
|
||||
set -e
|
||||
|
||||
export PYTHONUNBUFFERED=1
|
||||
|
||||
# Function to log model status before each run
|
||||
check_models() {
|
||||
echo "📦 Checking models..."
|
||||
|
||||
# Paths updated to point to persistent root /models
|
||||
if [ -d "/models/.insightface/models/buffalo_l" ]; then
|
||||
echo " ✅ InsightFace Buffalo_L: present"
|
||||
else
|
||||
echo " ⬇️ InsightFace Buffalo_L: not found — will download on first run (~300MB)"
|
||||
echo " ⬇️ InsightFace Buffalo_L: downloading..."
|
||||
fi
|
||||
|
||||
if [ -d "/models/huggingface/hub" ] && [ "$(find /models/huggingface/hub -maxdepth 1 -type d 2>/dev/null | wc -l)" -gt 1 ]; then
|
||||
echo " ✅ HuggingFace models: present"
|
||||
else
|
||||
echo " ⬇️ HuggingFace models: not found — will download on first run (SigLIP ~1GB, YOLOv9c ~500MB)"
|
||||
fi
|
||||
|
||||
echo "🚀 Starting if-curator..."
|
||||
}
|
||||
|
||||
SCHEDULE="${CRON_SCHEDULE:-}"
|
||||
AUTO="${AUTO_MODE:-false}"
|
||||
|
||||
# Check if interactive mode is viable
|
||||
if [ "$AUTO" != "true" ] && [ ! -t 0 ]; then
|
||||
echo "❌ AUTO_MODE is not enabled and no TTY is attached."
|
||||
echo "❌ Set AUTO_MODE=true for headless/automated runs."
|
||||
echo "❌ AUTO_MODE must be true for non-interactive execution."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Always run once on startup
|
||||
echo "▶ Running on startup..."
|
||||
check_models
|
||||
uv run if-curator
|
||||
|
||||
if [ -z "$SCHEDULE" ]; then
|
||||
echo "▶ No CRON_SCHEDULE set — exiting"
|
||||
exit 0
|
||||
if [ -n "${CRON_SCHEDULE:-}" ]; then
|
||||
echo "▶ Switching to scheduled mode..."
|
||||
exec uv run python3 /app/scheduler.py
|
||||
else
|
||||
echo "▶ Running once..."
|
||||
uv run python -m if_curator.cli
|
||||
fi
|
||||
|
||||
echo "▶ CRON_SCHEDULE set to: $SCHEDULE"
|
||||
echo "▶ Switching to scheduled mode..."
|
||||
exec uv run python3 /app/scheduler.py
|
||||
|
||||
|
||||
Reference in New Issue
Block a user