Spaces:
Running
on
Zero
Running
on
Zero
added imagelslider and examples with caching
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ from torchvision.transforms import Compose
|
|
| 9 |
import cv2
|
| 10 |
from huggingface_hub import hf_hub_download
|
| 11 |
from safetensors.torch import load_file
|
|
|
|
| 12 |
import spaces
|
| 13 |
|
| 14 |
# Helper function to load model from Hugging Face
|
|
@@ -69,7 +70,7 @@ def process_image(image, model, device):
|
|
| 69 |
|
| 70 |
# Convert to a PIL image
|
| 71 |
depth_image = Image.fromarray(depth_colored_hwc)
|
| 72 |
-
return depth_image
|
| 73 |
|
| 74 |
# Gradio interface function with GPU support
|
| 75 |
@spaces.GPU
|
|
@@ -114,10 +115,11 @@ def gradio_interface(image):
|
|
| 114 |
iface = gr.Interface(
|
| 115 |
fn=gradio_interface,
|
| 116 |
inputs=gr.Image(type="pil"), # Only image input, no mode selection
|
| 117 |
-
outputs=
|
| 118 |
title="Depth Estimation Demo",
|
| 119 |
-
description="Upload an image to see the depth estimation results. Our model is running on GPU for faster processing."
|
| 120 |
-
|
|
|
|
| 121 |
|
| 122 |
# Launch the Gradio interface
|
| 123 |
iface.launch()
|
|
|
|
| 9 |
import cv2
|
| 10 |
from huggingface_hub import hf_hub_download
|
| 11 |
from safetensors.torch import load_file
|
| 12 |
+
from gradio_imageslider import ImageSlider
|
| 13 |
import spaces
|
| 14 |
|
| 15 |
# Helper function to load model from Hugging Face
|
|
|
|
| 70 |
|
| 71 |
# Convert to a PIL image
|
| 72 |
depth_image = Image.fromarray(depth_colored_hwc)
|
| 73 |
+
return image, depth_image
|
| 74 |
|
| 75 |
# Gradio interface function with GPU support
|
| 76 |
@spaces.GPU
|
|
|
|
| 115 |
iface = gr.Interface(
|
| 116 |
fn=gradio_interface,
|
| 117 |
inputs=gr.Image(type="pil"), # Only image input, no mode selection
|
| 118 |
+
outputs = ImageSlider(label="Depth slider", type="pil", slider_color="pink"), # Depth image out with a slider
|
| 119 |
title="Depth Estimation Demo",
|
| 120 |
+
description="Upload an image to see the depth estimation results. Our model is running on GPU for faster processing.",
|
| 121 |
+
examples=["maizi.jpg", "hair.jpg", "videoframe_10273.png", "videoframe_2168.png", "videoframe_3289.png"],
|
| 122 |
+
cache_examples=True,)
|
| 123 |
|
| 124 |
# Launch the Gradio interface
|
| 125 |
iface.launch()
|