add online demo
Browse files
app.py
CHANGED
|
@@ -1,12 +1,9 @@
|
|
| 1 |
-
import
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
| 4 |
if __name__ == '__main__':
|
| 5 |
-
import subprocess
|
| 6 |
-
import sys
|
| 7 |
-
def run_command(command):
|
| 8 |
-
subprocess.check_call([sys.executable, '-m'] + command.split(), shell=False)
|
| 9 |
-
|
| 10 |
# Install the package in editable mode
|
| 11 |
run_command("pip install -e .")
|
| 12 |
|
|
@@ -37,8 +34,6 @@ if __name__ == '__main__':
|
|
| 37 |
# Build frontend
|
| 38 |
run_command("bash scripts/build_frontend.sh")
|
| 39 |
|
| 40 |
-
|
| 41 |
-
|
| 42 |
# Change back to the previous directory
|
| 43 |
run_command("cd ..")
|
| 44 |
|
|
@@ -51,7 +46,8 @@ if __name__ == '__main__':
|
|
| 51 |
# Download the model checkpoint
|
| 52 |
run_command("curl -o ./checkpoints/sam_vit_b_01ec64.pth https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth")
|
| 53 |
|
| 54 |
-
|
|
|
|
| 55 |
|
| 56 |
parser = argparse.ArgumentParser(description='SEAGULL', formatter_class=argparse.RawTextHelpFormatter)
|
| 57 |
parser.add_argument('--model', help='path to seagull model', default='Zevin2023/SEAGULL-7B')
|
|
@@ -59,4 +55,4 @@ if __name__ == '__main__':
|
|
| 59 |
args = parser.parse_args()
|
| 60 |
|
| 61 |
demo = Main_ui(args).load_demo()
|
| 62 |
-
demo.launch(
|
|
|
|
| 1 |
+
import subprocess
|
| 2 |
+
import sys
|
| 3 |
+
def run_command(command):
|
| 4 |
+
subprocess.check_call([sys.executable, '-m'] + command.split(), shell=False)
|
| 5 |
+
|
| 6 |
if __name__ == '__main__':
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
# Install the package in editable mode
|
| 8 |
run_command("pip install -e .")
|
| 9 |
|
|
|
|
| 34 |
# Build frontend
|
| 35 |
run_command("bash scripts/build_frontend.sh")
|
| 36 |
|
|
|
|
|
|
|
| 37 |
# Change back to the previous directory
|
| 38 |
run_command("cd ..")
|
| 39 |
|
|
|
|
| 46 |
# Download the model checkpoint
|
| 47 |
run_command("curl -o ./checkpoints/sam_vit_b_01ec64.pth https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth")
|
| 48 |
|
| 49 |
+
import argparse
|
| 50 |
+
from demo.UI import Main_ui
|
| 51 |
|
| 52 |
parser = argparse.ArgumentParser(description='SEAGULL', formatter_class=argparse.RawTextHelpFormatter)
|
| 53 |
parser.add_argument('--model', help='path to seagull model', default='Zevin2023/SEAGULL-7B')
|
|
|
|
| 55 |
args = parser.parse_args()
|
| 56 |
|
| 57 |
demo = Main_ui(args).load_demo()
|
| 58 |
+
demo.launch()
|