optimize: run.sh

This commit is contained in:
源文雨 2023-08-27 01:21:18 +08:00
parent e15b37e0ca
commit edcff1df04
2 changed files with 39 additions and 28 deletions

34
run.sh Normal file → Executable file
View File

@ -9,11 +9,16 @@ elif [[ "$(uname)" != "Linux" ]]; then
exit 1 exit 1
fi fi
requirements_file="requirements.txt" if [ -d ".venv" ]; then
echo "Activate venv..."
source .venv/bin/activate
else
echo "Create venv..."
requirements_file="requirements.txt"
# Check if Python 3.8 is installed # Check if Python 3.8 is installed
if ! command -v python3.8 &> /dev/null; then if ! command -v python3 &> /dev/null; then
echo "Python 3.8 not found. Attempting to install..." echo "Python 3 not found. Attempting to install 3.8..."
if [[ "$(uname)" == "Darwin" ]] && command -v brew &> /dev/null; then if [[ "$(uname)" == "Darwin" ]] && command -v brew &> /dev/null; then
brew install python@3.8 brew install python@3.8
elif [[ "$(uname)" == "Linux" ]] && command -v apt-get &> /dev/null; then elif [[ "$(uname)" == "Linux" ]] && command -v apt-get &> /dev/null; then
@ -23,23 +28,30 @@ if ! command -v python3.8 &> /dev/null; then
echo "Please install Python 3.8 manually." echo "Please install Python 3.8 manually."
exit 1 exit 1
fi fi
fi fi
# Check if required packages are installed and install them if not python3 -m venv .venv
if [ -f "${requirements_file}" ]; then source .venv/bin/activate
installed_packages=$(python3.8 -m pip freeze)
# Check if required packages are installed and install them if not
if [ -f "${requirements_file}" ]; then
installed_packages=$(python3 -m pip freeze)
while IFS= read -r package; do while IFS= read -r package; do
[[ "${package}" =~ ^#.* ]] && continue [[ "${package}" =~ ^#.* ]] && continue
package_name=$(echo "${package}" | sed 's/[<>=!].*//') package_name=$(echo "${package}" | sed 's/[<>=!].*//')
if ! echo "${installed_packages}" | grep -q "${package_name}"; then if ! echo "${installed_packages}" | grep -q "${package_name}"; then
echo "${package_name} not found. Attempting to install..." echo "${package_name} not found. Attempting to install..."
python3.8 -m pip install --upgrade "${package}" python3 -m pip install --upgrade "${package}"
fi fi
done < "${requirements_file}" done < "${requirements_file}"
else else
echo "${requirements_file} not found. Please ensure the requirements file with required packages exists." echo "${requirements_file} not found. Please ensure the requirements file with required packages exists."
exit 1 exit 1
fi
fi fi
# Download models
./tools/dlmodels.sh
# Run the main script # Run the main script
python3.8 infer-web.py --pycmd python3.8 python3 infer-web.py --pycmd python3

View File

@ -524,7 +524,7 @@ else
fi fi
echo checking $hb echo checking $hb
if [ -f "./pretrained/$hb" ]; then if [ -f "./$hb" ]; then
echo $hb in ./pretrained checked. echo $hb in ./pretrained checked.
else else
echo failed. starting download from huggingface. echo failed. starting download from huggingface.
@ -543,4 +543,3 @@ else
fi fi
echo required files check finished. echo required files check finished.
read -p "Press any key to continue..." -n1 -s