Update app.py
Browse files
app.py
CHANGED
|
@@ -204,7 +204,7 @@ def _default_render_cameras(batch_size: int = 1):
|
|
| 204 |
render_cameras = _build_camera_standard(extrinsics, render_camera_intrinsics)
|
| 205 |
return render_cameras.unsqueeze(0).repeat(batch_size, 1, 1)
|
| 206 |
|
| 207 |
-
@spaces.GPU
|
| 208 |
def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export_mesh=False, export_video=True, fps=30):
|
| 209 |
image = preprocess_image(image, source_size).to(model_wrapper.device)
|
| 210 |
source_camera = _default_source_camera(batch_size=1).to(model_wrapper.device)
|
|
|
|
| 204 |
render_cameras = _build_camera_standard(extrinsics, render_camera_intrinsics)
|
| 205 |
return render_cameras.unsqueeze(0).repeat(batch_size, 1, 1)
|
| 206 |
|
| 207 |
+
# @spaces.GPU
|
| 208 |
def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export_mesh=False, export_video=True, fps=30):
|
| 209 |
image = preprocess_image(image, source_size).to(model_wrapper.device)
|
| 210 |
source_camera = _default_source_camera(batch_size=1).to(model_wrapper.device)
|