SEAGULL / app.py
Zevin2023's picture
Update app.py
0b875ea verified
raw
history blame
1.91 kB
import os
import sys
if __name__ == '__main__':
# Install the package in editable mode
os.system("pip install -e .")
os.system("pip install git+https://github.com/facebookresearch/segment-anything.git")
from zipfile import ZipFile
z = ZipFile("gradio.zip")
z.extractall()
os.chdir("gradio-bbox")
os.system("pip install -e .")
sys.path.append(os.getcwd())
os.chdir("..")
# 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")
# print("nvm installed OK.")
# # 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")
# print("pnpm installed OK.")
# # Clone the Gradio BBox repository
# os.system("git clone https://github.com/chencn2020/gradio-bbox.git")
# Build frontend
# os.system("bash gradio-bbox/scripts/build_frontend.sh")
# Change back to the previous directory
# os.system("cd ..")
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()