Update index.html

index.html

@@ -25,13 +25,23 @@ import gradio as gr
 
 transformers = await import_transformers_js()
 pipeline = transformers.pipeline
-depth_estimator = await pipeline('depth-estimation', 'Xenova/depth-anything-large-hf');
+depth_estimator = await pipeline('depth-estimation', 'Xenova/depth-anything-small-hf');
 
 
 async def estimate(input_image):
     output = await depth_estimator(as_url(input_image))
-    # print("blob:", output.depth.toBlob())
-    return output
+
+    depth_image = output["depth"].to_pil()
+
+    tensor = output["predicted_depth"]
+    tensor_data = {
+        "dims": tensor.dims,
+        "type": tensor.type,
+        "data": tensor.data,
+        "size": tensor.size,
+    }
+
+    return depth_image, tensor_data
 
 demo = gr.Interface(
     fn=estimate,
@@ -39,6 +49,7 @@ demo = gr.Interface(
         gr.Image(type="filepath")
     ],
     outputs=[
+        gr.Image(),
         gr.JSON(),
     ],
     examples=[