Update app.py

app.py

@@ -5,16 +5,18 @@ import numpy as np
 from PIL import Image
 from pathlib import Path
 from depth_viewer import depthviewer2html
-import cv2
 
 feature_extractor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
 model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
 
 def process_image(image_path):
+    if image_path is None:
+        return ""
+    
     image_path = Path(image_path)
     image = Image.open(image_path)
     
-    # if wider than 512 pixels let's resample to keep it performant on phones etc
+    # if wider than 512 pixels let's resample to keep it performant
     if image.size[0] > 512:
         image = image.resize((512, int(512 * image.size[1] / image.size[0])), Image.Resampling.LANCZOS)
     
@@ -36,21 +38,26 @@ def process_image(image_path):
     
     output = prediction.cpu().numpy()
     depth = (output * 255 / np.max(output)).astype('uint8')
-    h = depthviewer2html(image, depth)
-    return h  # No need for list wrapper with single output
+    
+    return depthviewer2html(image, depth)
 
-title = "3d Visualization of Depth Maps Generated using MiDaS"
-description = "Improved 3D interactive depth viewer using Three.js embedded in a Gradio app. For more details see the <a href='https://colab.research.google.com/drive/1l2l8U7Vhq9RnvV2tHyfhrXKNuHfmb4IP?usp=sharing'>Colab Notebook.</a>"
-examples = [["examples/owl1.jpg"],['examples/marsattacks.jpg'],['examples/kitten.jpg']]
+title = "3D Visualization of Depth Maps Generated using MiDaS"
+description = "Improved 3D interactive depth viewer using Three.js embedded in a Gradio app."
 
-iface = gr.Interface(fn=process_image,
-                     inputs=[gr.Image(type="filepath",label="Input Image")],
-                     outputs=[gr.HTML(label='Depth Viewer',elem_id='depth-viewer')],
-                     title=title,
-                     description=description,
-                     examples=examples,
-                     allow_flagging="never",
-                     cache_examples=False,
-                     css='#depth-viewer: {height:300px;}')
+with gr.Blocks(css="#depth-viewer { height: 600px; }") as demo:
+    gr.Markdown(f"# {title}")
+    gr.Markdown(description)
+    
+    with gr.Row():
+        input_image = gr.Image(type="filepath", label="Input Image")
+    
+    output_html = gr.HTML(label="Depth Viewer", elem_id="depth-viewer")
+    
+    input_image.change(fn=process_image, inputs=input_image, outputs=output_html)
+    
+    gr.Examples(
+        examples=[["examples/owl1.jpg"], ["examples/marsattacks.jpg"], ["examples/kitten.jpg"]],
+        inputs=input_image
+    )
 
-iface.launch(server_name="0.0.0.0", debug=True)
+demo.launch(server_name="0.0.0.0")