guangkaixu commited on
Commit
91c383d
·
verified ·
1 Parent(s): 17777da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -9
app.py CHANGED
@@ -331,9 +331,9 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
331
  interactive=False,
332
  )
333
 
334
- filenames = []
335
- filenames.extend(["normal_%d.jpg" %(i+1) for i in range(10)])
336
- example_folder = os.path.join(os.path.dirname(__file__), "./images")
337
  Examples(
338
  fn=process_pipe_normal,
339
  examples=[
@@ -346,7 +346,8 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
346
  # directory_name="examples_normal",
347
  cache_examples=False,
348
  )
349
-
 
350
  with gr.Tab("Dichotomous Segmentation"):
351
  with gr.Row():
352
  with gr.Column():
@@ -387,7 +388,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
387
 
388
  filenames = []
389
  filenames.extend(["dis_%d.jpg" %(i+1) for i in range(10)])
390
- example_folder = os.path.join(os.path.dirname(__file__), "./images")
391
  Examples(
392
  fn=process_pipe_dis,
393
  examples=[
@@ -551,7 +552,4 @@ def main():
551
 
552
 
553
  if __name__ == "__main__":
554
- main()
555
-
556
-
557
-
 
331
  interactive=False,
332
  )
333
 
334
+ filenames = ['normal_1.jpg','normal_10.jpg']
335
+ # filenames.extend(["normal_%d.jpg" %(i+1) for i in range(10)])
336
+ example_folder = "images"
337
  Examples(
338
  fn=process_pipe_normal,
339
  examples=[
 
346
  # directory_name="examples_normal",
347
  cache_examples=False,
348
  )
349
+
350
+
351
  with gr.Tab("Dichotomous Segmentation"):
352
  with gr.Row():
353
  with gr.Column():
 
388
 
389
  filenames = []
390
  filenames.extend(["dis_%d.jpg" %(i+1) for i in range(10)])
391
+ example_folder = "./images"
392
  Examples(
393
  fn=process_pipe_dis,
394
  examples=[
 
552
 
553
 
554
  if __name__ == "__main__":
555
+ main()