Update app.py

app.py

@@ -76,7 +76,7 @@ def process_depth(
         processing_res=processing_res,
         batch_size=1 if processing_res == 0 else 0,
         show_progress_bar=False,
-        mode=mode,
+        mode='depth',
     )
 
     depth_pred = pipe_out.pred_np
@@ -85,7 +85,7 @@ def process_depth(
     np.save(path_out_fp32, depth_pred)
     depth_colored.save(path_out_vis)
     
-    path_out_16bit = os.path.join(path_output_dir, f"{name_base}_%s_16bit.png" %mode)
+    path_out_16bit = os.path.join(path_output_dir, f"{name_base}_depth_16bit.png")
     depth_16bit = (depth_pred * 65535.0).astype(np.uint16)
     Image.fromarray(depth_16bit).save(path_out_16bit, mode="I;16")
 
@@ -113,7 +113,7 @@ def process_normal(
         processing_res=processing_res,
         batch_size=1 if processing_res == 0 else 0,
         show_progress_bar=False,
-        mode=mode,
+        mode='normal',
     )
 
     depth_pred = pipe_out.pred_np
@@ -148,7 +148,7 @@ def process_dis(
         processing_res=processing_res,
         batch_size=1 if processing_res == 0 else 0,
         show_progress_bar=False,
-        mode=mode,
+        mode='seg',
     )
 
     depth_pred = pipe_out.pred_np