script.py

"""Script to prepare the env for gradio app."""
import sys
import subprocess

subprocess.check_call([sys.executable, "-m", "pip", "install", "huggingface_hub==0.25.1"])
# subprocess.check_call([sys.executable, "-m", "pip", "install", "xformers==0.0.21"])
# subprocess.run(
#     "pip install flash-attn --no-build-isolation",
#     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
#     shell=True,
# )
subprocess.check_call([sys.executable, "app.py"])