justin-zk commited on
Commit
6da6c93
1 Parent(s): 429c56d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -466,8 +466,8 @@ description = """
466
  main = gr.Interface(
467
  fn=inference,
468
  inputs=[
469
- gr.Image(label="in context image", type='pil'),
470
- gr.Image(label="in context mask", type='pil'),
471
  gr.Image(label="test image1", type='pil'),
472
  gr.Image(label="test image2", type='pil'),
473
  ],
@@ -514,8 +514,8 @@ main_scribble = gr.Interface(
514
  main_finetune = gr.Interface(
515
  fn=inference_finetune,
516
  inputs=[
517
- gr.Image(label="in context image", type='pil'),
518
- gr.Image(label="in context mask", type='pil'),
519
  gr.Image(label="test image1", type='pil'),
520
  gr.Image(label="test image2", type='pil'),
521
  ],
@@ -537,9 +537,7 @@ demo = gr.Blocks()
537
  with demo:
538
  gr.TabbedInterface(
539
  [main, main_scribble, main_finetune],
540
- # [main, main_finetune],
541
- # ["Personalize-SAM", "Personalize-SAM-F"],
542
- ["Personalize-SAM", "Personalize-SAM-Scribble", "Personalize-SAM-F"],
543
  )
544
 
545
  demo.launch(enable_queue=False)
 
466
  main = gr.Interface(
467
  fn=inference,
468
  inputs=[
469
+ gr.Image(label="one-shot image", type='pil'),
470
+ gr.Image(label="one-shot mask", type='pil'),
471
  gr.Image(label="test image1", type='pil'),
472
  gr.Image(label="test image2", type='pil'),
473
  ],
 
514
  main_finetune = gr.Interface(
515
  fn=inference_finetune,
516
  inputs=[
517
+ gr.Image(label="one-shot image", type='pil'),
518
+ gr.Image(label="one-shot mask", type='pil'),
519
  gr.Image(label="test image1", type='pil'),
520
  gr.Image(label="test image2", type='pil'),
521
  ],
 
537
  with demo:
538
  gr.TabbedInterface(
539
  [main, main_scribble, main_finetune],
540
+ ["PerSAM", " PerSAM-Scribble", "PerSAM-F"],
 
 
541
  )
542
 
543
  demo.launch(enable_queue=False)