added last updated notice to page and try to fix dimension mismatch warning

app.py

@@ -11,6 +11,9 @@ import os
 import trimesh
 import time
 import timm  # Add this import
+import subprocess
+import cv2  # Add this import
+from datetime import datetime
 
 # Ensure timm is properly loaded
 print(f"Timm version: {timm.__version__}")
@@ -137,13 +140,15 @@ def predict_depth(input_image):
             depth = depth.squeeze()
 
         # Print debug information
-        print(f"Depth shape: {depth.shape}")
-        print(f"Image shape: {image.shape}")
+        print(f"Original depth shape: {depth.shape}")
+        print(f"Original image shape: {image.shape}")
+
+        # Resize depth to match image dimensions
+        image_height, image_width = image.shape[2], image.shape[3]
+        depth = cv2.resize(depth, (image_width, image_height), interpolation=cv2.INTER_LINEAR)
 
-        # Ensure depth and image have the same dimensions
-        if depth.shape != image.shape[2:]:
-            # Resize depth to match image dimensions
-            depth = np.resize(depth, image.shape[2:])
+        print(f"Resized depth shape: {depth.shape}")
+        print(f"Final image shape: {image.shape}")
 
         # No downsampling
         downscale_factor = 1
@@ -185,28 +190,38 @@ def predict_depth(input_image):
             os.remove(temp_file)
 
 # Create the Gradio interface with appropriate input and output components. 
+last_updated = get_last_commit_timestamp()
+
 iface = gr.Interface(
     fn=predict_depth,
     inputs=gr.Image(type="filepath"),
     outputs=[
-        gr.Image(type="filepath", label="Depth Map"),  # Displays the depth map image
-        gr.Textbox(label="Focal Length or Error Message"),  # Shows focal length or error messages
-        gr.File(label="Download Raw Depth Map (CSV)"),  # Allows downloading the raw depth data
-        gr.Model3D(label="View 3D Model"),  # For viewing the 3D model
-        gr.File(label="Download 3D Model (OBJ)")  # For downloading the 3D model
+        gr.Image(type="filepath", label="Depth Map"),
+        gr.Textbox(label="Focal Length or Error Message"),
+        gr.File(label="Download Raw Depth Map (CSV)"),
+        gr.Model3D(label="View 3D Model"),
+        gr.File(label="Download 3D Model (OBJ)")
     ],
     title="DepthPro Demo with 3D Visualization",
     description=(
         "An enhanced demo that creates a textured 3D model from the input image and depth map.\n\n"
-        "Forked from https://huggingface.co/spaces/akhaliq/depth-pro and model from https://huggingface.co/apple/DepthPro"
+        "Forked from https://huggingface.co/spaces/akhaliq/depth-pro and model from https://huggingface.co/apple/DepthPro\n"
         "**Instructions:**\n"
         "1. Upload an image.\n"
         "2. The app will predict the depth map, display it, and provide the focal length.\n"
         "3. Download the raw depth data as a CSV file.\n"
         "4. View the generated 3D model textured with the original image.\n"
-        "5. Download the 3D model as an OBJ file if desired."
+        "5. Download the 3D model as an OBJ file if desired.\n\n"
+        f"Last updated: {last_updated}"
     ),
 )
 
 # Launch the Gradio interface with sharing enabled
-iface.launch(share=True)  # share=True allows you to share the interface with others.
+iface.launch(share=True)  # share=True allows you to share the interface with others.
+
+def get_last_commit_timestamp():
+    try:
+        timestamp = subprocess.check_output(['git', 'log', '-1', '--format=%cd', '--date=iso']).decode('utf-8').strip()
+        return datetime.fromisoformat(timestamp).strftime("%Y-%m-%d %H:%M:%S")
+    except Exception:
+        return "Unknown"

requirements.txt

@@ -6,4 +6,5 @@ torchvision
 pillow_heif==0.8.0
 timm
 trimesh==3.22.1
-gradio
+gradio
+opencv-python==4.5.5.64