SEAGULL / app.py
Zevin2023's picture
add online demo
0d95aff
raw
history blame
2.09 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 ..")
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()