justin-zk commited on
Commit
68318f4
1 Parent(s): 1b8ce63

update code

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -486,11 +486,11 @@ main = gr.Interface(
486
  ]
487
  )
488
 
489
-
490
  main_scribble = gr.Interface(
491
  fn=inference_scribble,
492
  inputs=[
493
- gr.ImageMask(label="[Stroke] Draw on Image", brush_radius=4),
494
  gr.Image(label="test image1"),
495
  gr.Image(label="test image2"),
496
  ],
@@ -508,7 +508,7 @@ main_scribble = gr.Interface(
508
  ["./examples/duck_toy_00.jpg", "./examples/duck_toy_01.jpg", "./examples/duck_toy_02.jpg"],
509
  ]
510
  )
511
-
512
 
513
  main_finetune = gr.Interface(
514
  fn=inference_finetune,
@@ -535,10 +535,10 @@ main_finetune = gr.Interface(
535
  demo = gr.Blocks()
536
  with demo:
537
  gr.TabbedInterface(
538
- [main, main_scribble, main_finetune],
539
- # [main, main_finetune],
540
- # ["Personalize-SAM", "Personalize-SAM-F"],
541
- ["Personalize-SAM", "Personalize-SAM-Scribble", "Personalize-SAM-F"],
542
  )
543
 
544
  demo.launch(enable_queue=False)
 
486
  ]
487
  )
488
 
489
+ """
490
  main_scribble = gr.Interface(
491
  fn=inference_scribble,
492
  inputs=[
493
+ gr.ImageMask(label="[Stroke] Draw on Image"),
494
  gr.Image(label="test image1"),
495
  gr.Image(label="test image2"),
496
  ],
 
508
  ["./examples/duck_toy_00.jpg", "./examples/duck_toy_01.jpg", "./examples/duck_toy_02.jpg"],
509
  ]
510
  )
511
+ """
512
 
513
  main_finetune = gr.Interface(
514
  fn=inference_finetune,
 
535
  demo = gr.Blocks()
536
  with demo:
537
  gr.TabbedInterface(
538
+ # [main, main_scribble, main_finetune],
539
+ [main, main_finetune],
540
+ ["Personalize-SAM", "Personalize-SAM-F"],
541
+ # ["Personalize-SAM", "Personalize-SAM-Scribble", "Personalize-SAM-F"],
542
  )
543
 
544
  demo.launch(enable_queue=False)