RefDiffNet / setup.sh
ShashwatGupta23's picture
Fix Gradio 4.49 pil images and HF launch (no share)
4dfed21
Raw History Blame
545 Bytes
#!/usr/bin/env bash
# Hugging Face Spaces: install vendored custom Ultralytics before app starts.
set -euo pipefail
cd "$(dirname "$0")"
# HF base image may install huggingface_hub 1.x; Gradio 4.x needs <1.0 (HfFolder)
pip install -q "huggingface_hub>=0.23.0,<1.0"
export YOLO_CONFIG_DIR="${YOLO_CONFIG_DIR:-/tmp/Ultralytics}"
mkdir -p "${YOLO_CONFIG_DIR}"
if [[ -d vendor ]]; then
pip install -q -e ./vendor
echo "Installed custom ultralytics from ./vendor"
else
echo "WARN: vendor/ missing — run prepare_space.sh before deploying"
fi