Update app.py

app.py

@@ -16,7 +16,6 @@ from typing import List, Optional, Tuple
 import cv2
 import gradio as gr
 import numpy as np
-import open3d as o3d
 import torch
 import torchvision.transforms as T
 import trimesh
@@ -102,11 +101,12 @@ def load_model(checkpoint_path: str = "checkpoints"):
 def save_obj(vertices: np.ndarray, faces: np.ndarray, colors: np.ndarray, filepath: str):
     vertices = vertices @ np.array([[1, 0, 0], [0, 1, 0], [0, 0, -1]])
     faces = faces[:, [2, 1, 0]]
-    mesh = o3d.geometry.TriangleMesh()
-    mesh.vertices = o3d.utility.Vector3dVector(vertices)
-    mesh.triangles = o3d.utility.Vector3iVector(faces)
-    mesh.vertex_colors = o3d.utility.Vector3dVector(colors / 255.0)
-    o3d.io.write_triangle_mesh(filepath, mesh, write_vertex_normals=False)
+    mesh = trimesh.Trimesh(
+        vertices=vertices,
+        faces=faces,
+        vertex_colors=colors.astype(np.uint8),
+    )
+    mesh.export(filepath)
 
 
 def position_to_c2w(position: Tuple[float, float, float]) -> torch.Tensor:
@@ -495,7 +495,7 @@ def build_demo():
             Upload a satellite image to **generate a 3D mesh** or **render a walkthrough video**.
 
             📌 **Input requirements:** The satellite image should be at **zoom level 20**
-            (same as the [VIGOR](https://github.com/Jeff-Zilence/VIGOR) dataset), resized to **512×512** pixels.
+            (same as the [VIGOR](https://github.com/Jeff-Zilence/VIGOR) dataset), then will be resized to the input size.
             You can download satellite tiles at this zoom level from any map tile API (e.g. Google Maps, Bing Maps, Mapbox).
             """
         )