justin-zk commited on
Commit
26c5293
·
1 Parent(s): 2a13d43

add main scribble func

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -499,7 +499,7 @@ main = gr.Interface(
499
  ]
500
  )
501
 
502
- """
503
  main_scribble = gr.Interface(
504
  fn=inference_scribble,
505
  inputs=[
@@ -520,7 +520,7 @@ main_scribble = gr.Interface(
520
  ["./examples/duck_toy_00.jpg", "./examples/duck_toy_01.jpg", "./examples/duck_toy_02.jpg"],
521
  ]
522
  )
523
- """
524
 
525
  main_finetune = gr.Interface(
526
  fn=inference_finetune,
@@ -546,10 +546,10 @@ main_finetune = gr.Interface(
546
  demo = gr.Blocks()
547
  with demo:
548
  gr.TabbedInterface(
549
- # [main, main_scribble, main_finetune],
550
- [main, main_finetune],
551
- ["Personalize-SAM", "Personalize-SAM-F"],
552
- # ["Personalize-SAM", "Personalize-SAM-Scribble", "Personalize-SAM-F"],
553
  )
554
 
555
  demo.launch()
 
499
  ]
500
  )
501
 
502
+
503
  main_scribble = gr.Interface(
504
  fn=inference_scribble,
505
  inputs=[
 
520
  ["./examples/duck_toy_00.jpg", "./examples/duck_toy_01.jpg", "./examples/duck_toy_02.jpg"],
521
  ]
522
  )
523
+
524
 
525
  main_finetune = gr.Interface(
526
  fn=inference_finetune,
 
546
  demo = gr.Blocks()
547
  with demo:
548
  gr.TabbedInterface(
549
+ [main, main_scribble, main_finetune],
550
+ # [main, main_finetune],
551
+ # ["Personalize-SAM", "Personalize-SAM-F"],
552
+ ["Personalize-SAM", "Personalize-SAM-Scribble", "Personalize-SAM-F"],
553
  )
554
 
555
  demo.launch()