Update app.py

app.py

@@ -67,8 +67,8 @@ def process_image(
     print(f"Processing image {name_base}{name_ext}")
 
     path_output_dir = tempfile.mkdtemp()
-    path_out_fp32 = os.path.join(path_output_dir, f"{name_base}_depth_fp32.npy")
-    path_out_vis = os.path.join(path_output_dir, f"{name_base}_depth_colored.png")
+    path_out_fp32 = os.path.join(path_output_dir, f"{name_base}_%s_fp32.npy" %mode)
+    path_out_vis = os.path.join(path_output_dir, f"{name_base}_%s_colored.png" %mode)
 
     input_image = Image.open(path_input)
 
@@ -87,7 +87,7 @@ def process_image(
     depth_colored.save(path_out_vis)
     
     if mode == 'depth':
-        path_out_16bit = os.path.join(path_output_dir, f"{name_base}_depth_16bit.png")
+        path_out_16bit = os.path.join(path_output_dir, f"{name_base}_%s_16bit.png" %mode)
         depth_16bit = (depth_pred * 65535.0).astype(np.uint16)
         Image.fromarray(depth_16bit).save(path_out_16bit, mode="I;16")
     else: