gr.interface() for predict_depth_normal

app.py

@@ -335,7 +335,7 @@ with gr.Blocks(title=title, css=custom_css) as demo:
     submit_button.click(fn=predict_depth_normal, inputs=[image_input, model_dropdown, fx, fy, state_cache], outputs=[depth_output, depth_file, normal_output, normal_file, state_cache, message_box])
     reconstruct_button.click(fn=unprojection_pcd, inputs=[state_cache], outputs=[pcd_output, pcd_ply, message_box])
 
-demo.launch()
+demo.launch(server_name="0.0.0.0")
 
 
 # iface = gr.Interface(
@@ -356,4 +356,26 @@ demo.launch()
 #     analytics_enabled=False
 # )
 
-# iface.launch()
+# iface.launch()
+
+
+iface = gr.Interface(
+    fn=predict_depth_normal,
+    inputs=[
+        gr.Image(type='pil', label="Original Image"),
+        gr.Dropdown(["vit-small", "vit-large"], label="Model"),
+        gr.Number(1000.0, label="fx in pixels"),
+        gr.Number(1000.0, label="fy in pixels")
+    ],
+    outputs=[
+        gr.Image(label="Output Depth"),
+        gr.File(label="Depth (.npy)"),
+        gr.Image(label="Output Normal"),
+        gr.File(label="Normal (.npy)"),
+        gr.Textbox(label="Messages")
+    ],
+    title="Metric3D",
+    description="Metric3Dv2: A versatile monocular geometric foundation model for zero-shot metric depth and surface normal estimation."
+)
+
+iface.launch()