ttengwang commited on
Commit
79c9f11
1 Parent(s): 9a84ec8
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -20,6 +20,9 @@ from segment_anything import sam_model_registry
20
 
21
 
22
  args = parse_augment()
 
 
 
23
  if args.segmenter_checkpoint is None:
24
  _, segmenter_checkpoint = prepare_segmenter(args.segmenter)
25
  else:
@@ -540,4 +543,4 @@ def create_ui():
540
  if __name__ == '__main__':
541
  iface = create_ui()
542
  iface.queue(concurrency_count=5, api_open=False, max_size=10)
543
- iface.launch(server_name="0.0.0.0", enable_queue=True, server_port=args.port, share=args.gradio_share)
 
20
 
21
 
22
  args = parse_augment()
23
+ args.segmenter = "huge"
24
+ args.segmenter_checkpoint = "sam_vit_h_4b8939.pth"
25
+
26
  if args.segmenter_checkpoint is None:
27
  _, segmenter_checkpoint = prepare_segmenter(args.segmenter)
28
  else:
 
543
  if __name__ == '__main__':
544
  iface = create_ui()
545
  iface.queue(concurrency_count=5, api_open=False, max_size=10)
546
+ iface.launch(server_name="0.0.0.0", enable_queue=True)