Spaces:
Running
Running
""" | |
#!/bin/bash | |
CUDA_ENABLED=${CUDA_ENABLED:-true} | |
DEVICE="" | |
if [ "${CUDA_ENABLED}" != "true" ]; then | |
DEVICE="--device cpu" | |
fi | |
exec python tools/webui.py ${DEVICE} | |
""" | |
import os | |
CUDA_ENABLED = os.environ.get("CUDA_ENABLED", "true") | |
DEVICE = "" | |
if CUDA_ENABLED != "true": | |
DEVICE = "--device cpu" | |
os.system(f"python tools/webui.py {DEVICE}") |