inital version

app.py

@@ -84,6 +84,28 @@ def run(input_file, resolution):
     vis_depth, depth_min, depth_max = visualize_depth(depth, ret_minmax=True)
     vis_prompt_depth = visualize_depth(prompt_depth[0, 0].detach().cpu().numpy(), depth_min=depth_min, depth_max=depth_max)
     vis_prompt_depth = cv2.resize(vis_prompt_depth, (vis_depth.shape[1], vis_depth.shape[0]), interpolation=cv2.INTER_NEAREST)
+    # Add text to vis_prompt_depth
+    text_x = vis_prompt_depth.shape[1] - 250 + 15
+    text_y = vis_prompt_depth.shape[0] - 45 + 27
+    vis_prompt_depth = cv2.rectangle(vis_prompt_depth, 
+                                     (vis_prompt_depth.shape[1] - 250, vis_prompt_depth.shape[0] - 45), 
+                                     (vis_prompt_depth.shape[1] - 5, vis_prompt_depth.shape[0] - 5), 
+                                     (70, 70, 70), -1)
+    vis_prompt_depth = cv2.putText(vis_prompt_depth, 'Prompt depth', 
+                                   (text_x, text_y), 
+                                   cv2.FONT_HERSHEY_SIMPLEX, 
+                                   1, (255, 255, 255), 2, cv2.LINE_AA)
+
+    text_x = 5 + 15
+    text_y = vis_depth.shape[0] - 45 + 27
+    vis_depth = cv2.rectangle(vis_depth, 
+                              (5, vis_depth.shape[0] - 45), 
+                              (250, vis_depth.shape[0] - 5), 
+                              (70, 70, 70), -1)
+    vis_depth = cv2.putText(vis_depth, 'Output depth', 
+                            (text_x, text_y), 
+                            cv2.FONT_HERSHEY_SIMPLEX, 
+                            1, (255, 255, 255), 2, cv2.LINE_AA)
 
     # PLY File
     ixt_path = os.path.join(input_dir, f'camera_matrix.csv')
@@ -138,7 +160,7 @@ Project Page: [Prompt Depth Anything](https://promptda.github.io/)
 Note:
 - Currently, this demo only supports inference for the first frame. If you need to obtain all depth frames, please refer to our [GitHub repo](https://github.com/DepthAnything/PromptDA).
 - The depth map is stored as uint16, with a unit of millimeters.
-- **You can refer to the bottom of this page for a example demo.**
+- **You can refer to the bottom of this page for an example demo.**
 """
 
 def main():
@@ -189,7 +211,7 @@ def main():
                          inputs=[input_file, resolution], 
                          outputs=outputs)
 
-    demo.launch()
+    demo.launch(share=True)
 # def main():
 #     gr.Interface(
 #         fn=run,