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 ..") | |
import subprocess | |
import sys | |
subprocess.check_call([sys.executable, 'pip', 'install', 'git+https://github.com/facebookresearch/segment-anything.git']) | |
subprocess.check_call([sys.executable, 'curl', '-o', './checkpoints/sam_vit_b_01ec64.pth', 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth']) | |
# 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() |