inital version

app.py

@@ -13,9 +13,8 @@ import gradio as gr
 from gradio_imageslider import ImageSlider
 import cv2
 import numpy as np
-import click
 import imageio
-from promptda.promptda import PromptDA
+# from promptda.promptda import PromptDA
 from promptda.utils.io_wrapper import load_image, load_depth
 from promptda.utils.depth_utils import visualize_depth, unproject_depth
 # import torch
@@ -23,7 +22,7 @@ DEVICE = 'cuda'
 # if torch.cuda.is_available(
 # ) else 'mps' if torch.backends.mps.is_available() else 'cpu'
 # model = PromptDA.from_pretrained('depth-anything/promptda_vitl').to(DEVICE).eval()
-model = PromptDA.from_pretrained('depth-anything/promptda_vitl').eval()
+# model = PromptDA.from_pretrained('depth-anything/promptda_vitl').eval()
 thread_pool_executor = ThreadPoolExecutor(max_workers=1)
 
 def delete_later(path: Union[str, os.PathLike], delay: int = 300):
@@ -144,76 +143,76 @@ Note:
 
 # @click.command()
 # @click.option('--share', is_flag=True, help='Whether to run the app in shared mode.')
-# def main():
-#     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-#         gr.Markdown(DESCRIPTION)
+def main():
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+        gr.Markdown(DESCRIPTION)
 
-#         with gr.Row():
-#             input_file = gr.File(type="filepath", label="Stray scanner app capture zip file")
-#             resolution = gr.Dropdown(choices=['756x1008', '1428x1904'], value='756x1008', label="Inference resolution")
-#             submit_btn = gr.Button("Submit")
+        with gr.Row():
+            input_file = gr.File(type="filepath", label="Stray scanner app capture zip file")
+            resolution = gr.Dropdown(choices=['756x1008', '1428x1904'], value='756x1008', label="Inference resolution")
+            submit_btn = gr.Button("Submit")
         
-#         # gr.Examples(examples=[
-#         #         ["data/assets/example0_chair.zip", "756x1008"]
-#         #     ],
-#         #     inputs=[input_file, resolution],
-#         #     label="Examples",
-#         # ) 
-
-#         with gr.Row():
-#             with gr.Column():
-#                 output_rgb = gr.Image(type="numpy", label="RGB Image")
-#             with gr.Column():
-#                 output_depths = ImageSlider(label="Output depth / prompt depth", position=0.5)
+        # gr.Examples(examples=[
+        #         ["data/assets/example0_chair.zip", "756x1008"]
+        #     ],
+        #     inputs=[input_file, resolution],
+        #     label="Examples",
+        # ) 
+
+        with gr.Row():
+            with gr.Column():
+                output_rgb = gr.Image(type="numpy", label="RGB Image")
+            with gr.Column():
+                output_depths = ImageSlider(label="Output depth / prompt depth", position=0.5)
         
-#         with gr.Row():
-#             with gr.Column():
-#                 output_3d_model = gr.Model3D(label="3D Viewer", display_mode='solid', clear_color=[1.0, 1.0, 1.0, 1.0])
-#             with gr.Column():
-#                 output_ply = gr.File(type="filepath", label="Download the unprojected point cloud as .ply file", height=30)
-#                 output_depth_map = gr.File(type="filepath", label="Download the depth map as .png file", height=30)
-#         outputs = [
-#             output_rgb,
-#             output_depths,
-#             output_3d_model,
-#             output_ply,
-#             output_depth_map,
-#         ]
-#         gr.Examples(examples=[
-#                 ["data/assets/example0_chair.zip", "756x1008"]
-#             ],
-#             fn=run,
-#             inputs=[input_file, resolution],
-#             outputs=outputs,
-#             label="Examples",
-#             cache_examples=True,
-#         ) 
-#         submit_btn.click(run, 
-#                          inputs=[input_file, resolution], 
-#                          outputs=outputs)
-
-#     demo.launch()
-def main():
-    gr.Interface(
-        fn=run,
-        inputs=[
-            gr.File(type="filepath", label="Stray scanner app capture zip file"),
-            gr.Dropdown(choices=['756x1008', '1428x1904'], value='756x1008', label="Inference resolution")
-        ],
-        outputs=[
-            gr.Image(type="numpy", label="RGB Image"),
-            ImageSlider(label="Depth map / prompt depth", position=0.5),
-            gr.Model3D(label="3D Viewer", display_mode='solid', clear_color=[1.0, 1.0, 1.0, 1.0]),
-            gr.File(type="filepath", label="Download the unprojected point cloud as .ply file"),
-            gr.File(type="filepath", label="Download the depth map as .png file"),
-        ],
-        title=None,
-        description=DESCRIPTION,
-        clear_btn=None,
-        allow_flagging="never",
-        theme=gr.themes.Soft(),
-        examples=[
-            ["data/assets/example0_chair.zip"]
+        with gr.Row():
+            with gr.Column():
+                output_3d_model = gr.Model3D(label="3D Viewer", display_mode='solid', clear_color=[1.0, 1.0, 1.0, 1.0])
+            with gr.Column():
+                output_ply = gr.File(type="filepath", label="Download the unprojected point cloud as .ply file", height=30)
+                output_depth_map = gr.File(type="filepath", label="Download the depth map as .png file", height=30)
+        outputs = [
+            output_rgb,
+            output_depths,
+            output_3d_model,
+            output_ply,
+            output_depth_map,
         ]
-    ).launch()
+        gr.Examples(examples=[
+                ["data/assets/example0_chair.zip", "756x1008"]
+            ],
+            fn=run,
+            inputs=[input_file, resolution],
+            outputs=outputs,
+            label="Examples",
+            cache_examples=True,
+        ) 
+        submit_btn.click(run, 
+                         inputs=[input_file, resolution], 
+                         outputs=outputs)
+
+    demo.launch()
+# def main():
+#     gr.Interface(
+#         fn=run,
+#         inputs=[
+#             gr.File(type="filepath", label="Stray scanner app capture zip file"),
+#             gr.Dropdown(choices=['756x1008', '1428x1904'], value='756x1008', label="Inference resolution")
+#         ],
+#         outputs=[
+#             gr.Image(type="numpy", label="RGB Image"),
+#             ImageSlider(label="Depth map / prompt depth", position=0.5),
+#             gr.Model3D(label="3D Viewer", display_mode='solid', clear_color=[1.0, 1.0, 1.0, 1.0]),
+#             gr.File(type="filepath", label="Download the unprojected point cloud as .ply file"),
+#             gr.File(type="filepath", label="Download the depth map as .png file"),
+#         ],
+#         title=None,
+#         description=DESCRIPTION,
+#         clear_btn=None,
+#         allow_flagging="never",
+#         theme=gr.themes.Soft(),
+#         examples=[
+#             ["data/assets/example0_chair.zip"]
+#         ]
+#     ).launch()
 main()