SEAGULL / app.py
Zevin2023's picture
add online demo
effd827
raw
history blame
1.98 kB
import os
if __name__ == '__main__':
# Install the package in editable mode
os.system("pip install -e .")
# Install NVM (Node Version Manager)
os.system("curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.3/install.sh | bash")
# Source the appropriate shell configuration file
os.system("source ~/.bashrc") # You can change to ~/.zshrc based on your shell
# Install Node.js version 18.16.0
os.system("nvm install v18.16.0")
# Install pnpm (package manager)
os.system("curl -fsSL https://get.pnpm.io/install.sh | sh -")
# Source the shell configuration file again (for pnpm)
os.system("source ~/.bashrc") # You can change to ~/.zshrc based on your shell
# Verify if pnpm was installed correctly
os.system("pnpm --version")
# Clone the Gradio BBox repository
os.system("git clone https://github.com/chencn2020/gradio-bbox.git")
# Change into the cloned repository directory
os.system("cd gradio-bbox")
# Build frontend
os.system("bash scripts/build_frontend.sh")
# Change back to the previous directory
os.system("cd ..")
# Install the package again in editable mode
os.system("pip install -e .")
# Install Segment Anything repository from GitHub
os.system("pip install git+https://github.com/facebookresearch/segment-anything.git")
# Download the model checkpoint
os.system("curl -o ./checkpoints/sam_vit_b_01ec64.pth https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth")
import argparse
from demo.UI import Main_ui
parser = argparse.ArgumentParser(description='SEAGULL', formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument('--model', help='path to seagull model', default='Zevin2023/SEAGULL-7B')
parser.add_argument('--example_path', help='path to examples', default='./imgs/Examples')
args = parser.parse_args()
demo = Main_ui(args).load_demo()
demo.launch()