Spaces:
mashroo
/
Running on Zero

YoussefAnso commited on
Commit
d454202
·
1 Parent(s): d63ad23

Update imports in inference.py to include zipfile and nvdiffrast for enhanced mesh processing capabilities and improved functionality.

Browse files
Files changed (1) hide show
  1. inference.py +4 -3
inference.py CHANGED
@@ -2,12 +2,13 @@ import numpy as np
2
  import torch
3
  import time
4
  import tempfile
5
- import os
6
- from PIL import Image
7
- import trimesh
8
  from util.utils import get_tri
9
  from mesh import Mesh
10
  from util.renderer import Renderer
 
11
 
12
 
13
  def generate3d(model, rgb, ccm, device):
 
2
  import torch
3
  import time
4
  import tempfile
5
+ import zipfile
6
+ import nvdiffrast.torch as dr
7
+
8
  from util.utils import get_tri
9
  from mesh import Mesh
10
  from util.renderer import Renderer
11
+ from kiui.mesh_utils import clean_mesh
12
 
13
 
14
  def generate3d(model, rgb, ccm, device):