toshas commited on
Commit
3636bd3
·
1 Parent(s): ca43de2

fix examples signature

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -36,9 +36,9 @@ def find_first_png(directory):
36
  return None
37
 
38
 
39
- def marigold_process(path_input, path_out_png=None, path_out_obj=None, path_out_2_png=None):
40
- if path_out_png is not None and path_out_obj is not None and path_out_2_png is not None:
41
- return path_out_png, path_out_obj, path_out_2_png
42
 
43
  path_input_dir = path_input + ".input"
44
  path_output_dir = path_input + ".output"
@@ -54,12 +54,12 @@ def marigold_process(path_input, path_out_png=None, path_out_obj=None, path_out_
54
  f"--denoise_steps 10 "
55
  )
56
 
57
- path_out_colored = find_first_png(path_output_dir + "/depth_colored")
58
- assert path_out_colored is not None, "Processing failed"
59
- path_out_bw = find_first_png(path_output_dir + "/depth_bw")
60
- assert path_out_bw is not None, "Processing failed"
61
 
62
- return path_out_colored, path_out_bw
63
 
64
 
65
  iface = gr.Interface(
 
36
  return None
37
 
38
 
39
+ def marigold_process(path_input, path_out_vis=None, path_out_pred=None):
40
+ if path_out_vis is not None and path_out_pred is not None:
41
+ return path_out_vis, path_out_pred
42
 
43
  path_input_dir = path_input + ".input"
44
  path_output_dir = path_input + ".output"
 
54
  f"--denoise_steps 10 "
55
  )
56
 
57
+ path_out_vis = find_first_png(path_output_dir + "/depth_colored")
58
+ assert path_out_vis is not None, "Processing failed"
59
+ path_out_pred = find_first_png(path_output_dir + "/depth_bw")
60
+ assert path_out_pred is not None, "Processing failed"
61
 
62
+ return path_out_vis, path_out_pred
63
 
64
 
65
  iface = gr.Interface(