JeffreyXiang commited on
Commit
0ed9085
·
1 Parent(s): 523e08e
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +2 -0
app.py CHANGED
@@ -33,6 +33,7 @@ def end_session(req: gr.Request):
33
  shutil.rmtree(user_dir)
34
 
35
 
 
36
  def preprocess_image(image: Image.Image) -> Image.Image:
37
  """
38
  Preprocess the input image.
@@ -181,6 +182,8 @@ def extract_glb(
181
  aabb=[[-0.5, -0.5, -0.5], [0.5, 0.5, 0.5]],
182
  decimation_target=decimation_target,
183
  texture_size=texture_size,
 
 
184
  use_tqdm=True,
185
  )[0]
186
  now = datetime.now()
 
33
  shutil.rmtree(user_dir)
34
 
35
 
36
+ @spaces.GPU()
37
  def preprocess_image(image: Image.Image) -> Image.Image:
38
  """
39
  Preprocess the input image.
 
182
  aabb=[[-0.5, -0.5, -0.5], [0.5, 0.5, 0.5]],
183
  decimation_target=decimation_target,
184
  texture_size=texture_size,
185
+ remesh=True,
186
+ remesh_band=1,
187
  use_tqdm=True,
188
  )[0]
189
  now = datetime.now()
requirements.txt CHANGED
@@ -12,6 +12,8 @@ opencv-python-headless==4.12.0.88
12
  trimesh==4.10.1
13
  transformers==4.57.3
14
  zstandard==0.25.0
 
 
15
  git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
16
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
17
  https://github.com/JeffreyXiang/Storages/releases/download/Space_Wheels_251210/cumesh-0.0.1-cp310-cp310-linux_x86_64.whl
 
12
  trimesh==4.10.1
13
  transformers==4.57.3
14
  zstandard==0.25.0
15
+ kornia==0.8.2
16
+ timm==1.0.22
17
  git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
18
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
19
  https://github.com/JeffreyXiang/Storages/releases/download/Space_Wheels_251210/cumesh-0.0.1-cp310-cp310-linux_x86_64.whl